add throughput entry to log
This commit is contained in:
1 parent
82a565362c
commit
b2f0459542
2 files changed
+5
-2
No files matched your search
@@ -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
|
||||
|
||||
Reference in new issue
Block a user