Update NexToU.py

PengchengShi1220 · Mar 29, 2024 · 1d01968 · 1d01968
1 parent cee4e41
commit 1d01968
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/network_architecture/NexToU.py b/network_architecture/NexToU.py
@@ -27,12 +27,8 @@ def __init__(self,
                  dropout_op_kwargs: dict = None,
                  nonlin: Union[None, Type[torch.nn.Module]] = None,
                  nonlin_kwargs: dict = None,
-                 deep_supervision: bool = False,
-                 nonlin_first: bool = False
+                 deep_supervision: bool = False
                  ):
-        """
-        nonlin_first: if True you get conv -> nonlin -> norm. Else it's conv -> norm -> nonlin
-        """
         super().__init__()
         if isinstance(n_blocks_per_stage, int):
             n_blocks_per_stage = [n_blocks_per_stage] * n_stages