diff --git a/main.py b/main.py index 2531812..77cc943 100644 --- 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)