This commit is contained in:
parent
574b65e2c0
commit
4e59c138ea
|
|
@ -376,7 +376,6 @@ def main():
|
|||
deepspeed=args.deepspeed,
|
||||
dataloader_drop_last=True,
|
||||
dataloader_num_workers=0,
|
||||
dataloader_prefetch_factor=1,
|
||||
report_to=([] if args.report_to == "none" else [args.report_to]),
|
||||
bf16=args.bf16,
|
||||
fp16=(not args.bf16),
|
||||
|
|
|
|||
Loading…
Reference in New Issue