From 9923686187b3ab42990d2ed8f2fdcf633ccfb1cf Mon Sep 17 00:00:00 2001 From: tangy5 Date: Tue, 4 Apr 2023 22:07:52 -0700 Subject: [PATCH] remove umused imports Signed-off-by: tangy5 --- monai/networks/nets/universal_model.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/monai/networks/nets/universal_model.py b/monai/networks/nets/universal_model.py index 2bc0505ad6..e4e6c39236 100644 --- a/monai/networks/nets/universal_model.py +++ b/monai/networks/nets/universal_model.py @@ -2,8 +2,6 @@ import torch import torch.nn as nn -from model.text_embedding import TextEncoder -from model.head_controller import HeadController from monai.networks.blocks.text_embedding import TextEncoder from monai.networks.blocks.head_controller import HeadController @@ -104,16 +102,6 @@ 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)