tangger/openpi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

tweaking comment

Browse Source
This commit is contained in:
niccolofusai13
2025-03-02 05:50:52 +00:00
parent 92b1082442
commit 6c514a6d8a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/openpi/training/checkpoints.py
View File

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