Merge pull request #4066 from injet-zhou/main

add throughput entry to training log Former-commit-id: f2580ad403
2025-12-16 03:40:34 +08:00 · 2024-06-06 03:32:04 +08:00
parent d31c9c73c7 a2931b813b
commit 229794a148
2 changed files with 5 additions and 2 deletions
--- a/src/llamafactory/webui/runner.py
+++ b/src/llamafactory/webui/runner.py
@@ -132,6 +132,7 @@ class Runner:
            pure_bf16=(get("train.compute_type") == "pure_bf16"),
            plot_loss=True,
            ddp_timeout=180000000,
+            include_num_input_tokens_seen=True,
        )

        # checkpoints