Skip to content

Commit 942a717

Browse files
committed
fix LR scheduler patience param in train_schnet script
1 parent b35dda1 commit 942a717

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

nablaDFT/schnet/train_schnet.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020

2121
ReduceLROnPlateauFactor = 0.8
2222
ReduceLROnPlateauMinLR = 1e-6
23+
ReduceLROnPlateauPatience = 100
2324

2425

2526
def seed_everything(seed=42):
@@ -135,7 +136,7 @@ def __init__(
135136

136137
scheduler_args = {
137138
"factor": ReduceLROnPlateauFactor,
138-
"patience": args.patience,
139+
"patience": ReduceLROnPlateauPatience,
139140
"min_lr": ReduceLROnPlateauMinLR,
140141
}
141142

0 commit comments

Comments
 (0)