Skip to content

Commit 827a3d6

Browse files
committed
Add current checkpoints output back to CheckpointSaver (via logger)
1 parent 63961b3 commit 827a3d6

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

timm/utils.py

+5
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,11 @@ def save_checkpoint(self, model, optimizer, args, epoch, model_ema=None, metric=
6565
self.checkpoint_files, key=lambda x: x[1],
6666
reverse=not self.decreasing) # sort in descending order if a lower metric is not better
6767

68+
checkpoints_str = "Current checkpoints:\n"
69+
for c in self.checkpoint_files:
70+
checkpoints_str += ' {}\n'.format(c)
71+
logging.info(checkpoints_str)
72+
6873
if metric is not None and (self.best_metric is None or self.cmp(metric, self.best_metric)):
6974
self.best_epoch = epoch
7075
self.best_metric = metric

0 commit comments

Comments
 (0)