From cb1402cfe50aeeb63587d56c12c8a3c73a8bf9d9 Mon Sep 17 00:00:00 2001 From: tangy5 Date: Tue, 4 Apr 2023 23:24:04 -0700 Subject: [PATCH] add args Signed-off-by: tangy5 --- monai/networks/nets/universal_model.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/monai/networks/nets/universal_model.py b/monai/networks/nets/universal_model.py index e4e6c39236..01d546c084 100644 --- a/monai/networks/nets/universal_model.py +++ b/monai/networks/nets/universal_model.py @@ -102,6 +102,16 @@ def __init__( img_size: Union[Sequence[int], int], in_channels: int, out_channels: int, + depths: Sequence[int] = (2, 2, 2, 2), + num_heads: Sequence[int] = (3, 6, 12, 24), + feature_size: int = 48, + norm_name: Union[Tuple, str] = "instance", + drop_rate: float = 0.0, + attn_drop_rate: float = 0.0, + dropout_path_rate: float = 0.0, + normalize: bool = True, + use_checkpoint: bool = False, + spatial_dims: int = 3, ): super(SwinUNETR_backbone, self).__init__(img_size,in_channels,out_channels,feature_size=48)