From c5ae100a47e112484a4a75bc2ef13c078bac78ed Mon Sep 17 00:00:00 2001 From: Sander Vandenhaute Date: Sun, 21 Jul 2024 13:11:43 -0400 Subject: [PATCH] fix wrong swa default --- psiflow/models/_mace.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/psiflow/models/_mace.py b/psiflow/models/_mace.py index c761f6a..ea7d770 100644 --- a/psiflow/models/_mace.py +++ b/psiflow/models/_mace.py @@ -89,8 +89,8 @@ class MACEConfig: lr_factor: float = 0.8 scheduler_patience: int = 50 lr_scheduler_gamma: float = 0.9993 - swa: bool = True # default: False - start_swa: Optional[int] = 0 # never start swa + swa: bool = False # default: False + start_swa: Optional[int] = int(1e10) # never start swa ema: bool = False ema_decay: float = 0.99 max_num_epochs: int = 2048