Add **kwargs to LayerNormNC constructor

rwth-i6 · Nov 5, 2024 · 6fa0d2a · 6fa0d2a
1 parent 5aa74f8
commit 6fa0d2a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/i6_models/parts/conformer/norm.py b/i6_models/parts/conformer/norm.py
@@ -9,11 +9,11 @@ class LayerNormNC(nn.LayerNorm):
     see here: https://pytorch.org/docs/stable/generated/torch.nn.LayerNorm.html
     """
 
-    def __init__(self, channels: int):
+    def __init__(self, channels: int, **kwargs):
         """
         :param channels: number of channels for normalization
         """
-        super().__init__(channels)
+        super().__init__(channels, **kwargs)
 
     def forward(self, tensor: torch.Tensor) -> torch.Tensor:
         """