add throughput entry to log

Former-commit-id: 691f999f64c7bac78761e4354f89816d2f0d46fc
This commit is contained in:
faddddeout
2024-06-04 11:04:29 +00:00
parent 2ac2cde03e
commit 99ecb0daaf
2 changed files with 5 additions and 2 deletions

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