.

2025-08-26 20:08:31 +08:00 · 2025-08-26 20:08:31 +08:00 · c8fad0d1da
parent 619bebcf57
commit c8fad0d1da
1 changed files with 1 additions and 3 deletions
--- a/train_sft_ds.py
+++ b/train_sft_ds.py
@ -634,9 +634,7 @@ def main():
        save_steps=args.save_steps,
        save_total_limit=2,
        deepspeed=(args.deepspeed if args.deepspeed and os.path.isfile(args.deepspeed) else None),
-        # dataloader_drop_last=False,          # 关键：别丢尾，避免空 batch
-        dataloader_drop_last=True,          # 关键：别丢尾，避免空 batch
-        dispatch_batches=False,
+        dataloader_drop_last=False,          # 关键：别丢尾，避免空 batch
        dataloader_num_workers=0,
        dataloader_prefetch_factor=None,
        dataloader_pin_memory=False,