Increase checkpoint frequency from 1000 to 5000

msosav · Jan 11, 2025 · aadfc04 · aadfc04
1 parent 99031ab
commit aadfc04
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -19,7 +19,7 @@
     }
 
     callback = CheckpointAndLoggingCallback(
-        check_freq=1000, save_path=config["checkpoint_dir"]
+        check_freq=5000, save_path=config["checkpoint_dir"]
     )
 
     env = PreprocessEnv(config)