Fix config parameters.

mila-iqia · Jun 6, 2024 · 69d1225 · 69d1225
1 parent 4a3c10d
commit 69d1225
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/mila_cluster/diffusion/config_diffusion_mace.yaml b/examples/mila_cluster/diffusion/config_diffusion_mace.yaml
@@ -29,7 +29,8 @@ model:
     interaction_cls_first: RealAgnosticInteractionBlock
     num_interactions: 2
     hidden_irreps: 8x0e + 8x1o
-    MLP_irreps: 8x0e
+    mlp_irreps: 8x0e
+
     avg_num_neighbors: 1
     correlation: 3
     gate: silu