Reverted local changes to schnet

elliottower · Apr 19, 2021 · 7e14c66 · 7e14c66
1 parent 9fce6b5
commit 7e14c66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/configs/s2ef/200k/schnet/schnet.yml b/configs/s2ef/200k/schnet/schnet.yml
@@ -11,8 +11,8 @@ model:
   use_pbc: True
 
 optim:
-  batch_size: 50
-  eval_batch_size: 25
+  batch_size: 32
+  eval_batch_size: 16
   num_workers: 64
   lr_initial: 0.0005
   lr_gamma: 0.1