From 83dad35e8f68cd834a28ec012955e8df9722eca6 Mon Sep 17 00:00:00 2001 From: Balint Kovacs Date: Fri, 1 Mar 2024 10:50:32 +0100 Subject: [PATCH] compatibility with dynamic_network_architectures --- .../experiment_planners/resencUNet_planner.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nnunetv2/experiment_planning/experiment_planners/resencUNet_planner.py b/nnunetv2/experiment_planning/experiment_planners/resencUNet_planner.py index a7b2d2305..0ed95320c 100644 --- a/nnunetv2/experiment_planning/experiment_planners/resencUNet_planner.py +++ b/nnunetv2/experiment_planning/experiment_planners/resencUNet_planner.py @@ -2,7 +2,7 @@ from copy import deepcopy from typing import Union, List, Tuple -from dynamic_network_architectures.architectures.residual_unet import ResidualEncoderUNet +from dynamic_network_architectures.architectures.unet import ResidualEncoderUNet from dynamic_network_architectures.building_blocks.helper import convert_dim_to_conv_op, get_matching_instancenorm from torch import nn @@ -232,4 +232,4 @@ def _keygen(patch_size, strides): n_conv_per_stage_decoder=(1, 1, 1, 1, 1, 1), conv_bias=True, norm_op=nn.InstanceNorm2d, norm_op_kwargs={}, dropout_op=None, nonlin=nn.LeakyReLU, nonlin_kwargs={'inplace': True}, deep_supervision=True) - print(net.compute_conv_feature_map_size((512, 512))) # -> 129793792 \ No newline at end of file + print(net.compute_conv_feature_map_size((512, 512))) # -> 129793792