From a93a79568dcd16907ba75cc6de066361b2aad403 Mon Sep 17 00:00:00 2001
From: Sean McLellan <sean@baristalabs.io>
Date: Tue, 23 Aug 2022 21:56:05 -0400
Subject: [PATCH] Tweak save_top_k setting

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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