From 2cb0ac094267e45a2c5c76885969b998fb828ace Mon Sep 17 00:00:00 2001 From: hongbozheng <112345628+hongbozheng@users.noreply.github.com> Date: Tue, 9 Jul 2024 20:19:24 -0500 Subject: [PATCH] Corrected forward operations order and norm layer parameter --- GANDLF/models/seg_modules/DownsamplingModule.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/GANDLF/models/seg_modules/DownsamplingModule.py b/GANDLF/models/seg_modules/DownsamplingModule.py index 24c92ce77..10eded967 100644 --- 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)