diff --git a/utils/checkpoints.py b/utils/checkpoints.py
index 30db495..b564e40 100644
--- a/utils/checkpoints.py
+++ b/utils/checkpoints.py
@@ -39,5 +39,6 @@ def load_model(model: torch.nn.Module, checkpoint, optimizer: torch.nn.Module=No
     model.load_state_dict(checkpoint['model'])
     if optimizer is not None: optimizer.load_state_dict(checkpoint['optimizer'].state_dict())
     if scaler is not None and checkpoint['scaler'] is not None: scaler.load_state_dict(checkpoint['scaler'])
+    print(f'Loaded model from {checkpoints[-1]} at iteration {iteration}')
     return int(iteration) if iteration != '' else 0