Update main.py

agosztolai · Dec 6, 2023 · b6da468 · b6da468
1 parent d423a6b
commit b6da468
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/MARBLE/main.py b/MARBLE/main.py
@@ -413,7 +413,7 @@ def load_model(self, loadpath):
         self._epoch = checkpoint["epoch"]
         self.load_state_dict(checkpoint["model_state_dict"])
         self.optimizer_state_dict = checkpoint["optimizer_state_dict"]
-        if hatattr(self, 'losses'):
+        if hasattr(self, 'losses'):
             self.losses = checkpoint["losses"]
 
     def save_model(self, optimizer, losses, outdir=None, best=False, timestamp=""):