From eee75b10ec9db2b7ee865ca6ab29562667f814ff Mon Sep 17 00:00:00 2001 From: YunLiu <55491388+KumoLiu@users.noreply.github.com> Date: Mon, 2 Dec 2024 11:25:42 +0800 Subject: [PATCH] fix #8248 Signed-off-by: YunLiu <55491388+KumoLiu@users.noreply.github.com> --- monai/networks/nets/swin_unetr.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monai/networks/nets/swin_unetr.py b/monai/networks/nets/swin_unetr.py index 32b817d584..77f0d2ec2f 100644 --- a/monai/networks/nets/swin_unetr.py +++ b/monai/networks/nets/swin_unetr.py @@ -75,7 +75,7 @@ def __init__( dropout_path_rate: float = 0.0, normalize: bool = True, norm_layer: type[LayerNorm] = nn.LayerNorm, - patch_norm: bool = True, + patch_norm: bool = False, use_checkpoint: bool = False, spatial_dims: int = 3, downsample: str | nn.Module = "merging", @@ -102,7 +102,7 @@ def __init__( dropout_path_rate: drop path rate. normalize: normalize output intermediate features in each stage. norm_layer: normalization layer. - patch_norm: whether to apply normalization to the patch embedding. + patch_norm: whether to apply normalization to the patch embedding. Default is False. use_checkpoint: use gradient checkpointing for reduced memory usage. spatial_dims: number of spatial dims. downsample: module used for downsampling, available options are `"mergingv2"`, `"merging"` and a