correct order of if

asappresearch · Dec 15, 2020 · 7e4c758 · 7e4c758
1 parent e9d6ed2
commit 7e4c758
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sru/modules.py b/sru/modules.py
@@ -151,9 +151,9 @@ def __init__(self,
         self.layer_norm: Optional[nn.Module]= None
         if layer_norm:
             if post_ln:
-                self.layer_norm = nn.LayerNorm(self.input_size)
-            else:
                 self.layer_norm = nn.LayerNorm(self.output_size)
+            else:
+                self.layer_norm = nn.LayerNorm(self.input_size)
 
         self.reset_parameters()