Corrected forward operations order and norm layer parameter

mlcommons · Jul 10, 2024 · 2cb0ac0 · 2cb0ac0
1 parent b604142
commit 2cb0ac0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GANDLF/models/seg_modules/DownsamplingModule.py b/GANDLF/models/seg_modules/DownsamplingModule.py
@@ -39,7 +39,7 @@ def __init__(
         if act_kwargs is None:
             act_kwargs = {"negative_slope": 1e-2, "inplace": True}
 
-        self.in_0 = norm(output_channels, **norm_kwargs)
+        self.in_0 = norm(input_channels, **norm_kwargs)
 
         self.conv0 = conv(input_channels, output_channels, **conv_kwargs)