From 607904b64f7a5f983a9e2c8ac1eaa170a9b9683d Mon Sep 17 00:00:00 2001 From: YunLiu <55491388+KumoLiu@users.noreply.github.com> Date: Mon, 27 May 2024 11:03:18 +0800 Subject: [PATCH] Fix #7802 Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com> --- monai/networks/nets/resnet.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/monai/networks/nets/resnet.py b/monai/networks/nets/resnet.py index 2cd7c8102a..6e61db07ca 100644 --- a/monai/networks/nets/resnet.py +++ b/monai/networks/nets/resnet.py @@ -94,13 +94,12 @@ def __init__( super().__init__() conv_type: Callable = Conv[Conv.CONV, spatial_dims] - norm_layer = get_norm_layer(name=norm, spatial_dims=spatial_dims, channels=planes) self.conv1 = conv_type(in_planes, planes, kernel_size=3, padding=1, stride=stride, bias=False) - self.bn1 = norm_layer + self.bn1 = get_norm_layer(name=norm, spatial_dims=spatial_dims, channels=planes) self.act = get_act_layer(name=act) self.conv2 = conv_type(planes, planes, kernel_size=3, padding=1, bias=False) - self.bn2 = norm_layer + self.bn2 = get_norm_layer(name=norm, spatial_dims=spatial_dims, channels=planes) self.downsample = downsample self.stride = stride