File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -75,7 +75,7 @@ def __init__(
7575 dropout_path_rate : float = 0.0 ,
7676 normalize : bool = True ,
7777 norm_layer : type [LayerNorm ] = nn .LayerNorm ,
78- patch_norm : bool = True ,
78+ patch_norm : bool = False ,
7979 use_checkpoint : bool = False ,
8080 spatial_dims : int = 3 ,
8181 downsample : str | nn .Module = "merging" ,
@@ -102,7 +102,7 @@ def __init__(
102102 dropout_path_rate: drop path rate.
103103 normalize: normalize output intermediate features in each stage.
104104 norm_layer: normalization layer.
105- patch_norm: whether to apply normalization to the patch embedding.
105+ patch_norm: whether to apply normalization to the patch embedding. Default is False.
106106 use_checkpoint: use gradient checkpointing for reduced memory usage.
107107 spatial_dims: number of spatial dims.
108108 downsample: module used for downsampling, available options are `"mergingv2"`, `"merging"` and a
You can’t perform that action at this time.
0 commit comments