Changed default save frequency advances #17

pab1s · May 28, 2024 · b3a1779 · b3a1779
1 parent 7abf4f9
commit b3a1779
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/callbacks/checkpoint.py b/callbacks/checkpoint.py
@@ -15,7 +15,7 @@ class Checkpoint(Callback):
         verbose (bool, optional): Whether to print the checkpoint save path. Default is False.
     """
 
-    def __init__(self, checkpoint_dir, model, optimizer, scheduler=None, save_freq=1, verbose=False):
+    def __init__(self, checkpoint_dir, model, optimizer, scheduler=None, save_freq=5, verbose=False):
         self.checkpoint_dir = checkpoint_dir
         self.model = model
         self.optimizer = optimizer