From 729f9d37a5947ec6228b575560bc3494ead4d5ca Mon Sep 17 00:00:00 2001
From: Sander Vandenhaute <svdenhau@gmail.com>
Date: Tue, 9 Jul 2024 05:13:31 -0400
Subject: [PATCH] use start_swa=0 to avoid mace tools.train to attempt starting
 swa when it shouldnt

---
 psiflow/models/_mace.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/psiflow/models/_mace.py b/psiflow/models/_mace.py
index 598b361..f02c66a 100644
--- a/psiflow/models/_mace.py
+++ b/psiflow/models/_mace.py
@@ -90,7 +90,7 @@ class MACEConfig:
     scheduler_patience: int = 50
     lr_scheduler_gamma: float = 0.9993
     swa: bool = True  # default: False
-    start_swa: Optional[int] = None  # never start swa
+    start_swa: Optional[int] = 0  # never start swa
     ema: bool = False
     ema_decay: float = 0.99
     max_num_epochs: int = 2048