Skip to content

Commit 6c514a6

Browse files
tweaking comment
1 parent 92b1082 commit 6c514a6

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/openpi/training/checkpoints.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -48,8 +48,8 @@ def initialize_checkpoint_dir(
4848
),
4949
)
5050

51-
# special case: the checkpoint directory exists and the user requests to resume training, but the training run did
52-
# not get to the first checkpoint saved. in this case, we don't actually want the train script to try and restore a
51+
# Special case: the checkpoint directory exists and the user requests to resume training, but the training run did
52+
# not get to the first checkpoint saved. In this case, we don't actually want the train script to try and restore a
5353
# checkpoint, since it will fail.
5454
if resuming and tuple(mngr.all_steps()) in [(), (0,)]:
5555
logging.info("Checkpoint directory exists, but does not contain any checkpoints. Aborting resume.")

0 commit comments

Comments
 (0)