hailin
/
jd_train
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

.

This commit is contained in:
hailin 2025-08-25 20:00:54 +08:00
parent 574b65e2c0
commit 4e59c138ea
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train_sft_ds.py
View File

@ -376,7 +376,6 @@ def main():
deepspeed=args.deepspeed, deepspeed=args.deepspeed,
dataloader_drop_last=True, dataloader_drop_last=True,
dataloader_num_workers=0, dataloader_num_workers=0,
dataloader_prefetch_factor=1,
report_to=([] if args.report_to == "none" else [args.report_to]), report_to=([] if args.report_to == "none" else [args.report_to]),
bf16=args.bf16, bf16=args.bf16,
fp16=(not args.bf16), fp16=(not args.bf16),