Skip to content

Commit

Permalink
remove umused imports
Browse files Browse the repository at this point in the history
Signed-off-by: tangy5 <[email protected]>
  • Loading branch information
tangy5 committed Apr 5, 2023
1 parent 2b7b04b commit 9923686
Showing 1 changed file with 0 additions and 12 deletions.
12 changes: 0 additions & 12 deletions monai/networks/nets/universal_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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)

Expand Down

0 comments on commit 9923686

Please sign in to comment.