From b387e9d64dedfbb2a207d153ffc08bb459c8817b Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sun, 2 Jul 2023 14:59:46 +0200 Subject: [PATCH] add logging to loeading checkpoints --- utils/checkpoints.py | 1 + 1 file changed, 1 insertion(+) 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