Skip to content

Commit

Permalink
use start_swa=0 to avoid mace tools.train to attempt starting swa whe…
Browse files Browse the repository at this point in the history
…n it shouldnt
  • Loading branch information
svandenhaute committed Jul 9, 2024
1 parent 185a940 commit 729f9d3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion psiflow/models/_mace.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 729f9d3

Please sign in to comment.