Skip to content

Commit 9e5bbd9

Browse files
fix: saving ckpt error when ckpt folder has some folder inside (#3516)
1 parent cef9589 commit 9e5bbd9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

nerfstudio/engine/trainer.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -478,8 +478,8 @@ def save_checkpoint(self, step: int) -> None:
478478
)
479479
# possibly delete old checkpoints
480480
if self.config.save_only_latest_checkpoint:
481-
# delete everything else in the checkpoint folder
482-
for f in self.checkpoint_dir.glob("*"):
481+
# delete every other checkpoint in the checkpoint folder
482+
for f in self.checkpoint_dir.glob("*.ckpt"):
483483
if f != ckpt_path:
484484
f.unlink()
485485

0 commit comments

Comments
 (0)