tes removing simplify

ACEsuit · Oct 1, 2024 · 264817c · 264817c
1 parent 24bf8a6
commit 264817c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mace/modules/blocks.py b/mace/modules/blocks.py
@@ -625,7 +625,7 @@ def _setup(self) -> None:
         )
 
         # Linear
-        irreps_mid = irreps_mid.simplify()
+        # irreps_mid = irreps_mid.simplify()
         self.irreps_out = self.target_irreps
         self.linear = o3.Linear(
             irreps_mid, self.irreps_out, internal_weights=True, shared_weights=True