fix: ivy BatchNorm2D _extra_repr

ivy-llc · May 29, 2024 · 827459d · 827459d
1 parent 586299e
commit 827459d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ivy/stateful/norms.py b/ivy/stateful/norms.py
@@ -212,8 +212,8 @@ def _forward(self, inputs):
 
     def _extra_repr(self) -> str:
         return (
-            f"num_features={self._num_features}, affine={self._affine}, "
-            f"data_format={self._data_format}, epsilon={self._epsilon} "
+            f"num_features={self.num_features}, affine={self._affine}, "
+            f"data_format={self.data_format}, epsilon={self._epsilon} "
             f"momentum={self._momentum}, "
             f"track_running_stats={self._track_running_stats}"
         )