avoid getting undefined

huggingface · Oct 23, 2023 · 7a63691 · 7a63691
1 parent 89ba0da
commit 7a63691
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/timm/models/_helpers.py b/timm/models/_helpers.py
@@ -136,8 +136,8 @@ def resume_checkpoint(
                 if 'version' in checkpoint and checkpoint['version'] > 1:
                     resume_epoch += 1  # start at the next epoch, old checkpoints incremented before save
 
-            if log_info:
-                _logger.info("Loaded checkpoint '{}' (epoch {})".format(checkpoint_path, checkpoint['epoch']))
+                if log_info:
+                    _logger.info("Loaded checkpoint '{}' (epoch {})".format(checkpoint_path, checkpoint['epoch']))
         else:
             model.load_state_dict(checkpoint)
             if log_info: