switch component

ACEsuit · Oct 1, 2024 · fa86566 · fa86566
1 parent 1899b0b
commit fa86566
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mace/modules/blocks.py b/mace/modules/blocks.py
@@ -571,7 +571,7 @@ def _setup(self) -> None:
             affine=False,
             reduce="mean",
             instance=True,
-            normalization="norm",
+            normalization="component",
         )
 
     def forward(
@@ -653,7 +653,7 @@ def _setup(self) -> None:
             affine=False,
             reduce="mean",
             instance=True,
-            normalization="norm",
+            normalization="component",
         )
 
     def forward(