update

Oneflow-Inc · Sep 19, 2024 · 195cdb9 · 195cdb9
1 parent 078a1db
commit 195cdb9
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/projects/Qwen/configs/qwen_sft.py b/projects/Qwen/configs/qwen_sft.py
@@ -64,11 +64,11 @@
         output_dir="./sft_result",
         train_micro_batch_size=1,
         test_micro_batch_size=1,
-        train_epoch=3,
+        train_epoch=1,
         train_iter=1,
-        log_period=10,
+        log_period=1,
         warmup_ratio=1 / 3,
-        num_accumulation_steps=8,
+        num_accumulation_steps=1,
         rdma_enabled=False,
         amp=dict(enabled=True),
         activation_checkpoint=dict(enabled=True),