Update config.yaml

globus-labs · Oct 3, 2023 · e34d124 · e34d124
1 parent 22ee59b
commit e34d124
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/config.yaml b/config.yaml
@@ -13,7 +13,7 @@ model: egnn_dynamics
 lr: 2.0e-4
 batch_size: 64
 n_layers: 6
-n_epochs: 1500
+n_epochs: 3
 test_epochs: 20
 n_stability_samples: 10
 nf: 128