Swap the try and except blocks for ModelCheckpoint

HHousen · Feb 2, 2022 · 686047e · 686047e
1 parent 8310971
commit 686047e
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main.py b/src/main.py
@@ -92,16 +92,17 @@ def main(args):
         try:
             args.checkpoint_callback = ModelCheckpoint(
                 save_top_k=-1,
-                period=1,
+                every_n_epochs=1,
                 verbose=True,
             )
         except TypeError:
             logger.warning(
-                "'period' parameter of ModelCheckpoint has been renamed to 'every_n_epochs'."
+                "'every_n_epochs' parameter of ModelCheckpoint is not found. "
+                + "Defaulting to its old name, 'period'."
             )
             args.checkpoint_callback = ModelCheckpoint(
                 save_top_k=-1,
-                every_n_epochs=1,
+                period=1,
                 verbose=True,
             )