diff --git a/main.py b/main.py index 0c611cd743..5653cf5e06 100644 --- a/main.py +++ b/main.py @@ -627,7 +627,7 @@ if __name__ == "__main__": if hasattr(model, "monitor"): print(f"Monitoring {model.monitor} as checkpoint metric.") default_modelckpt_cfg["params"]["monitor"] = model.monitor - default_modelckpt_cfg["params"]["save_top_k"] = 1 + default_modelckpt_cfg["params"]["save_top_k"] = 3 if "modelcheckpoint" in lightning_config: modelckpt_cfg = lightning_config.modelcheckpoint