Merge pull request #4066 from injet-zhou/main

add throughput entry to training log
This commit is contained in:
hoshi-hiyouga authored and GitHub committed 2024-06-06 03:32:04 +08:00
commit f2580ad403
2 files changed
+5 -2

No files matched your search

+1
View File
@@ -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