Skip to content

Commit 48e8598

Browse files
author
Orbax Authors
committed
Update docstring for train_steps in EmergencyCheckpointManagerBenchmarkConfig.
PiperOrigin-RevId: 819039352
1 parent 9a23bd1 commit 48e8598

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

checkpoint/orbax/checkpoint/_src/testing/benchmarks/emergency_checkpoint_manager_benchmark.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ class EcmBenchmarkOptions(benchmarks_core.BenchmarkOptions):
4949
saved.
5050
local_max_to_keep: The maximum number of local checkpoints to keep.
5151
replica_axis_index: The index of the replica axis in the global mesh.
52-
train_steps: The number of training steps to run.
52+
train_steps: The number of training steps to run. Added to test
5353
"""
5454

5555
persistent_save_interval_steps: int | Sequence[int] = 5

0 commit comments

Comments
 (0)