add throughput entry to log

Former-commit-id: b2f0459542
This commit is contained in:
faddddeout
2024-06-04 11:04:29 +00:00
parent 51e3229528
commit a2931b813b
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