Delete Dataset262/nnUNetTrainerMRCT_loss_masked_perception_masked__nnUNetResEncUNetLPlans__3d_fullres/fold_4

Dataset262/nnUNetTrainerMRCT_loss_masked_perception_masked__nnUNetResEncUNetLPlans__3d_fullres/fold_4/checkpoint_final.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:a53eb93ead6be5736e97b05407ffcc92829de1605140e0383b2c20a52916cd14
-size 815577005

Dataset262/nnUNetTrainerMRCT_loss_masked_perception_masked__nnUNetResEncUNetLPlans__3d_fullres/fold_4/debug.json

@@ -1,58 +0,0 @@
-{
-    "_best_ema": "None",
-    "aim_run": "<Run#-3472116665707052186 name=5d726a9fb5f2480ba80f2655 repo=<Repo#6177984172896394163 path=/datasets/work/hb-synthrad2023/work/synthrad2025/bw_workplace/data/nnunet_struct/results/runs_aim/.aim read_only=None>>",
-    "batch_size": "2",
-    "configuration_manager": "{'data_identifier': 'nnUNetPlans_3d_fullres', 'preprocessor_name': 'DefaultPreprocessor', 'batch_size': 2, 'patch_size': [56, 192, 160], 'median_image_size_in_voxels': [89.0, 296.0, 279.0], 'spacing': [3.0, 1.0, 1.0], 'normalization_schemes': ['ZScoreNormalization'], 'use_mask_for_norm': [False], 'resampling_fn_data': 'resample_data_or_seg_to_shape', 'resampling_fn_seg': 'resample_data_or_seg_to_shape', 'resampling_fn_data_kwargs': {'is_seg': False, 'order': 3, 'order_z': 0, 'force_separate_z': None}, 'resampling_fn_seg_kwargs': {'is_seg': True, 'order': 1, 'order_z': 0, 'force_separate_z': None}, 'resampling_fn_probabilities': 'resample_data_or_seg_to_shape', 'resampling_fn_probabilities_kwargs': {'is_seg': False, 'order': 1, 'order_z': 0, 'force_separate_z': None}, 'architecture': {'network_class_name': 'dynamic_network_architectures.architectures.unet.ResidualEncoderUNet', 'arch_kwargs': {'n_stages': 6, 'features_per_stage': [32, 64, 128, 256, 320, 320], 'conv_op': 'torch.nn.modules.conv.Conv3d', 'kernel_sizes': [[1, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3]], 'strides': [[1, 1, 1], [1, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2], [1, 2, 2]], 'n_blocks_per_stage': [1, 3, 4, 6, 6, 6], 'n_conv_per_stage_decoder': [1, 1, 1, 1, 1], 'conv_bias': True, 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm3d', 'norm_op_kwargs': {'eps': 1e-05, 'affine': True}, 'dropout_op': None, 'dropout_op_kwargs': None, 'nonlin': 'torch.nn.LeakyReLU', 'nonlin_kwargs': {'inplace': True}, 'deep_supervision': False}, '_kw_requires_import': ['conv_op', 'norm_op', 'dropout_op', 'nonlin']}, 'batch_dice': False}",
-    "configuration_name": "3d_fullres",
-    "cudnn_version": 90501,
-    "current_epoch": "0",
-    "dataloader_train": "<nnunetv2.training.data_augmentation.custom_transforms.limited_length_multithreaded_augmenter.LimitedLenWrapper object at 0x155230258a10>",
-    "dataloader_train.generator": "<nnunetv2.training.dataloading.data_loader_3d.nnUNetDataLoader3D_MRCT_mask object at 0x15523025a490>",
-    "dataloader_train.num_processes": "12",
-    "dataloader_train.transform": "Compose ( [RemoveLabelTransform( output_key = 'seg', input_key = 'seg', replace_with = 0, remove_label = -1 ), RenameTransform( delete_old = True, out_key = 'target', in_key = 'seg' ), NumpyToTensor( keys = ['data', 'target', 'mask'], cast_to = 'float' )] )",
-    "dataloader_val": "<nnunetv2.training.data_augmentation.custom_transforms.limited_length_multithreaded_augmenter.LimitedLenWrapper object at 0x15523056a0d0>",
-    "dataloader_val.generator": "<nnunetv2.training.dataloading.data_loader_3d.nnUNetDataLoader3D_MRCT_mask object at 0x15523056ab10>",
-    "dataloader_val.num_processes": "6",
-    "dataloader_val.transform": "Compose ( [RemoveLabelTransform( output_key = 'seg', input_key = 'seg', replace_with = 0, remove_label = -1 ), RenameTransform( delete_old = True, out_key = 'target', in_key = 'seg' ), NumpyToTensor( keys = ['data', 'target', 'mask'], cast_to = 'float' )] )",
-    "dataset_json": "{'labels': {'label_001': '1', 'background': 0}, 'channel_names': {'0': 'MR'}, 'numTraining': 221, 'file_ending': '.mha'}",
-    "debug_custom": "{'run_hash': '5d726a9fb5f2480ba80f2655'}",
-    "decoder_type": "standard",
-    "device": "cuda:0",
-    "disable_checkpointing": "False",
-    "enable_deep_supervision": "False",
-    "fold": "4",
-    "folder_with_segs_from_previous_stage": "None",
-    "gpu_name": "NVIDIA H100",
-    "grad_scaler": "<torch.amp.grad_scaler.GradScaler object at 0x15543daf2750>",
-    "hostname": "g032",
-    "image_loss_weight": "0.5",
-    "inference_allowed_mirroring_axes": "None",
-    "initial_lr": "0.01",
-    "is_cascaded": "False",
-    "is_ddp": "False",
-    "label_manager": "<nnunetv2.utilities.label_handling.label_handling.LabelManager object at 0x15543daf2690>",
-    "local_rank": "0",
-    "log_file": "/datasets/work/hb-synthrad2023/work/synthrad2025/bw_workplace/data/nnunet_struct/results/Dataset262_synthrad2025_task1_MR_HN_pre_v2r_stitched_masked/nnUNetTrainerMRCT_loss_masked_perception_masked__nnUNetResEncUNetLPlans__3d_fullres/fold_4/training_log_2025_8_13_01_37_31.txt",
-    "logger": "<nnunetv2.training.logging.nnunet_logger.nnUNetLogger object at 0x15543daf25d0>",
-    "loss": "SynPerceptionLoss(\n  (seg_model): ResidualEncoderUNet(\n    (encoder): ResidualEncoder(\n      (stem): StackedConvBlocks(\n        (convs): Sequential(\n          (0): ConvDropoutNormReLU(\n            (conv): Conv3d(1, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n            (norm): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n            (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n            (all_modules): Sequential(\n              (0): Conv3d(1, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n              (1): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n              (2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n          )\n        )\n      )\n      (stages): Sequential(\n        (0): StackedResidualBlocks(\n          (blocks): Sequential(\n            (0): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(32, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n                (norm): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(32, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n                  (1): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(32, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n                (norm): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(32, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n                  (1): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n          )\n        )\n        (1): StackedResidualBlocks(\n          (blocks): Sequential(\n            (0): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(32, 64, kernel_size=(3, 3, 3), stride=(1, 2, 2), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(32, 64, kernel_size=(3, 3, 3), stride=(1, 2, 2), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              (skip): Sequential(\n                (0): AvgPool3d(kernel_size=[1, 2, 2], stride=[1, 2, 2], padding=0)\n                (1): ConvDropoutNormReLU(\n                  (conv): Conv3d(32, 64, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                  (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(32, 64, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                    (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n              )\n            )\n            (1): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (2): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n          )\n        )\n        (2): StackedResidualBlocks(\n          (blocks): Sequential(\n            (0): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(64, 128, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(64, 128, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              (skip): Sequential(\n                (0): AvgPool3d(kernel_size=[2, 2, 2], stride=[2, 2, 2], padding=0)\n                (1): ConvDropoutNormReLU(\n                  (conv): Conv3d(64, 128, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                  (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(64, 128, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                    (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n              )\n            )\n            (1): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (2): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (3): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n          )\n        )\n        (3): StackedResidualBlocks(\n          (blocks): Sequential(\n            (0): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(128, 256, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(128, 256, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              (skip): Sequential(\n                (0): AvgPool3d(kernel_size=[2, 2, 2], stride=[2, 2, 2], padding=0)\n                (1): ConvDropoutNormReLU(\n                  (conv): Conv3d(128, 256, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(128, 256, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n              )\n            )\n            (1): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (2): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (3): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (4): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (5): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n          )\n        )\n        (4): StackedResidualBlocks(\n          (blocks): Sequential(\n            (0): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(256, 320, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(256, 320, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              (skip): Sequential(\n                (0): AvgPool3d(kernel_size=[2, 2, 2], stride=[2, 2, 2], padding=0)\n                (1): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 320, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 320, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n              )\n            )\n            (1): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (2): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (3): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (4): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (5): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n          )\n        )\n        (5): StackedResidualBlocks(\n          (blocks): Sequential(\n            (0): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 2, 2), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 2, 2), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              (skip): Sequential(\n                (0): AvgPool3d(kernel_size=[1, 2, 2], stride=[1, 2, 2], padding=0)\n              )\n            )\n            (1): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (2): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (3): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (4): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n            (5): BasicBlockD(\n              (conv1): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                )\n              )\n              (conv2): ConvDropoutNormReLU(\n                (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (all_modules): Sequential(\n                  (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                )\n              )\n              (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n            )\n          )\n        )\n      )\n    )\n    (decoder): UNetDecoder(\n      (encoder): ResidualEncoder(\n        (stem): StackedConvBlocks(\n          (convs): Sequential(\n            (0): ConvDropoutNormReLU(\n              (conv): Conv3d(1, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n              (norm): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n              (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n              (all_modules): Sequential(\n                (0): Conv3d(1, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n                (1): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n        )\n        (stages): Sequential(\n          (0): StackedResidualBlocks(\n            (blocks): Sequential(\n              (0): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(32, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n                  (norm): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(32, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n                    (1): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(32, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n                  (norm): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(32, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n                    (1): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n          (1): StackedResidualBlocks(\n            (blocks): Sequential(\n              (0): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(32, 64, kernel_size=(3, 3, 3), stride=(1, 2, 2), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(32, 64, kernel_size=(3, 3, 3), stride=(1, 2, 2), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n                (skip): Sequential(\n                  (0): AvgPool3d(kernel_size=[1, 2, 2], stride=[1, 2, 2], padding=0)\n                  (1): ConvDropoutNormReLU(\n                    (conv): Conv3d(32, 64, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                    (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (all_modules): Sequential(\n                      (0): Conv3d(32, 64, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                      (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    )\n                  )\n                )\n              )\n              (1): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (2): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(64, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n          (2): StackedResidualBlocks(\n            (blocks): Sequential(\n              (0): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(64, 128, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(64, 128, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n                (skip): Sequential(\n                  (0): AvgPool3d(kernel_size=[2, 2, 2], stride=[2, 2, 2], padding=0)\n                  (1): ConvDropoutNormReLU(\n                    (conv): Conv3d(64, 128, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                    (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (all_modules): Sequential(\n                      (0): Conv3d(64, 128, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                      (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    )\n                  )\n                )\n              )\n              (1): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (2): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (3): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(128, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n          (3): StackedResidualBlocks(\n            (blocks): Sequential(\n              (0): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(128, 256, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(128, 256, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n                (skip): Sequential(\n                  (0): AvgPool3d(kernel_size=[2, 2, 2], stride=[2, 2, 2], padding=0)\n                  (1): ConvDropoutNormReLU(\n                    (conv): Conv3d(128, 256, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                    (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (all_modules): Sequential(\n                      (0): Conv3d(128, 256, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                      (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    )\n                  )\n                )\n              )\n              (1): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (2): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (3): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (4): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (5): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n          (4): StackedResidualBlocks(\n            (blocks): Sequential(\n              (0): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(256, 320, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(256, 320, kernel_size=(3, 3, 3), stride=(2, 2, 2), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n                (skip): Sequential(\n                  (0): AvgPool3d(kernel_size=[2, 2, 2], stride=[2, 2, 2], padding=0)\n                  (1): ConvDropoutNormReLU(\n                    (conv): Conv3d(256, 320, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                    (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (all_modules): Sequential(\n                      (0): Conv3d(256, 320, kernel_size=(1, 1, 1), stride=(1, 1, 1), bias=False)\n                      (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    )\n                  )\n                )\n              )\n              (1): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (2): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (3): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (4): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (5): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n          (5): StackedResidualBlocks(\n            (blocks): Sequential(\n              (0): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 2, 2), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 2, 2), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n                (skip): Sequential(\n                  (0): AvgPool3d(kernel_size=[1, 2, 2], stride=[1, 2, 2], padding=0)\n                )\n              )\n              (1): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (2): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (3): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (4): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n              (5): BasicBlockD(\n                (conv1): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                    (2): LeakyReLU(negative_slope=0.01, inplace=True)\n                  )\n                )\n                (conv2): ConvDropoutNormReLU(\n                  (conv): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                  (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  (all_modules): Sequential(\n                    (0): Conv3d(320, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                    (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                  )\n                )\n                (nonlin2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n        )\n      )\n      (stages): ModuleList(\n        (0): StackedConvBlocks(\n          (convs): Sequential(\n            (0): ConvDropoutNormReLU(\n              (conv): Conv3d(640, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n              (norm): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n              (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n              (all_modules): Sequential(\n                (0): Conv3d(640, 320, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (1): InstanceNorm3d(320, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n        )\n        (1): StackedConvBlocks(\n          (convs): Sequential(\n            (0): ConvDropoutNormReLU(\n              (conv): Conv3d(512, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n              (norm): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n              (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n              (all_modules): Sequential(\n                (0): Conv3d(512, 256, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (1): InstanceNorm3d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n        )\n        (2): StackedConvBlocks(\n          (convs): Sequential(\n            (0): ConvDropoutNormReLU(\n              (conv): Conv3d(256, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n              (norm): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n              (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n              (all_modules): Sequential(\n                (0): Conv3d(256, 128, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (1): InstanceNorm3d(128, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n        )\n        (3): StackedConvBlocks(\n          (convs): Sequential(\n            (0): ConvDropoutNormReLU(\n              (conv): Conv3d(128, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n              (norm): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n              (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n              (all_modules): Sequential(\n                (0): Conv3d(128, 64, kernel_size=(3, 3, 3), stride=(1, 1, 1), padding=(1, 1, 1))\n                (1): InstanceNorm3d(64, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n        )\n        (4): StackedConvBlocks(\n          (convs): Sequential(\n            (0): ConvDropoutNormReLU(\n              (conv): Conv3d(64, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n              (norm): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n              (nonlin): LeakyReLU(negative_slope=0.01, inplace=True)\n              (all_modules): Sequential(\n                (0): Conv3d(64, 32, kernel_size=(1, 3, 3), stride=(1, 1, 1), padding=(0, 1, 1))\n                (1): InstanceNorm3d(32, eps=1e-05, momentum=0.1, affine=True, track_running_stats=False)\n                (2): LeakyReLU(negative_slope=0.01, inplace=True)\n              )\n            )\n          )\n        )\n      )\n      (transpconvs): ModuleList(\n        (0): ConvTranspose3d(320, 320, kernel_size=(1, 2, 2), stride=(1, 2, 2))\n        (1): ConvTranspose3d(320, 256, kernel_size=(2, 2, 2), stride=(2, 2, 2))\n        (2): ConvTranspose3d(256, 128, kernel_size=(2, 2, 2), stride=(2, 2, 2))\n        (3): ConvTranspose3d(128, 64, kernel_size=(2, 2, 2), stride=(2, 2, 2))\n        (4): ConvTranspose3d(64, 32, kernel_size=(1, 2, 2), stride=(1, 2, 2))\n      )\n      (seg_layers): ModuleList(\n        (0): Conv3d(320, 32, kernel_size=(1, 1, 1), stride=(1, 1, 1))\n        (1): Conv3d(256, 32, kernel_size=(1, 1, 1), stride=(1, 1, 1))\n        (2): Conv3d(128, 32, kernel_size=(1, 1, 1), stride=(1, 1, 1))\n        (3): Conv3d(64, 32, kernel_size=(1, 1, 1), stride=(1, 1, 1))\n        (4): Conv3d(32, 32, kernel_size=(1, 1, 1), stride=(1, 1, 1))\n      )\n    )\n  )\n  (L1): L1Loss()\n  (image_loss): myMaskedMSE(\n    (mse): myMSE()\n  )\n)",
-    "lr_scheduler": "<nnunetv2.training.lr_scheduler.polylr.PolyLRScheduler object at 0x15543e708b10>",
-    "my_init_kwargs": "{'plans': {'dataset_name': 'Dataset262_synthrad2025_task1_MR_HN_pre_v2r_stitched_masked', 'plans_name': 'nnUNetResEncUNetLPlans', 'original_median_spacing_after_transp': [3.0, 1.0, 1.0], 'original_median_shape_after_transp': [89, 296, 279], 'image_reader_writer': 'SimpleITKIO', 'transpose_forward': [0, 1, 2], 'transpose_backward': [0, 1, 2], 'configurations': {'2d': {'data_identifier': 'nnUNetPlans_2d', 'preprocessor_name': 'DefaultPreprocessor', 'batch_size': 30, 'patch_size': [320, 320], 'median_image_size_in_voxels': [296.0, 279.0], 'spacing': [1.0, 1.0], 'normalization_schemes': ['ZScoreNormalization'], 'use_mask_for_norm': [False], 'resampling_fn_data': 'resample_data_or_seg_to_shape', 'resampling_fn_seg': 'resample_data_or_seg_to_shape', 'resampling_fn_data_kwargs': {'is_seg': False, 'order': 3, 'order_z': 0, 'force_separate_z': None}, 'resampling_fn_seg_kwargs': {'is_seg': True, 'order': 1, 'order_z': 0, 'force_separate_z': None}, 'resampling_fn_probabilities': 'resample_data_or_seg_to_shape', 'resampling_fn_probabilities_kwargs': {'is_seg': False, 'order': 1, 'order_z': 0, 'force_separate_z': None}, 'architecture': {'network_class_name': 'dynamic_network_architectures.architectures.unet.ResidualEncoderUNet', 'arch_kwargs': {'n_stages': 7, 'features_per_stage': [32, 64, 128, 256, 512, 512, 512], 'conv_op': 'torch.nn.modules.conv.Conv2d', 'kernel_sizes': [[3, 3], [3, 3], [3, 3], [3, 3], [3, 3], [3, 3], [3, 3]], 'strides': [[1, 1], [2, 2], [2, 2], [2, 2], [2, 2], [2, 2], [2, 2]], 'n_blocks_per_stage': [1, 3, 4, 6, 6, 6, 6], 'n_conv_per_stage_decoder': [1, 1, 1, 1, 1, 1], 'conv_bias': True, 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm2d', 'norm_op_kwargs': {'eps': 1e-05, 'affine': True}, 'dropout_op': None, 'dropout_op_kwargs': None, 'nonlin': 'torch.nn.LeakyReLU', 'nonlin_kwargs': {'inplace': True}}, '_kw_requires_import': ['conv_op', 'norm_op', 'dropout_op', 'nonlin']}, 'batch_dice': True}, '3d_fullres': {'data_identifier': 'nnUNetPlans_3d_fullres', 'preprocessor_name': 'DefaultPreprocessor', 'batch_size': 2, 'patch_size': [56, 192, 160], 'median_image_size_in_voxels': [89.0, 296.0, 279.0], 'spacing': [3.0, 1.0, 1.0], 'normalization_schemes': ['ZScoreNormalization'], 'use_mask_for_norm': [False], 'resampling_fn_data': 'resample_data_or_seg_to_shape', 'resampling_fn_seg': 'resample_data_or_seg_to_shape', 'resampling_fn_data_kwargs': {'is_seg': False, 'order': 3, 'order_z': 0, 'force_separate_z': None}, 'resampling_fn_seg_kwargs': {'is_seg': True, 'order': 1, 'order_z': 0, 'force_separate_z': None}, 'resampling_fn_probabilities': 'resample_data_or_seg_to_shape', 'resampling_fn_probabilities_kwargs': {'is_seg': False, 'order': 1, 'order_z': 0, 'force_separate_z': None}, 'architecture': {'network_class_name': 'dynamic_network_architectures.architectures.unet.ResidualEncoderUNet', 'arch_kwargs': {'n_stages': 6, 'features_per_stage': [32, 64, 128, 256, 320, 320], 'conv_op': 'torch.nn.modules.conv.Conv3d', 'kernel_sizes': [[1, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3]], 'strides': [[1, 1, 1], [1, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2], [1, 2, 2]], 'n_blocks_per_stage': [1, 3, 4, 6, 6, 6], 'n_conv_per_stage_decoder': [1, 1, 1, 1, 1], 'conv_bias': True, 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm3d', 'norm_op_kwargs': {'eps': 1e-05, 'affine': True}, 'dropout_op': None, 'dropout_op_kwargs': None, 'nonlin': 'torch.nn.LeakyReLU', 'nonlin_kwargs': {'inplace': True}}, '_kw_requires_import': ['conv_op', 'norm_op', 'dropout_op', 'nonlin']}, 'batch_dice': False}}, 'experiment_planner_used': 'nnUNetPlannerResEncL', 'label_manager': 'LabelManager', 'foreground_intensity_properties_per_channel': {'0': {'max': 4142.0, 'mean': 75.25260162353516, 'median': 0.0, 'min': 0.0, 'percentile_00_5': 0.0, 'percentile_99_5': 1222.0, 'std': 197.03660583496094}}}, 'configuration': '3d_fullres', 'fold': 4, 'dataset_json': {'labels': {'label_001': '1', 'background': 0}, 'channel_names': {'0': 'MR'}, 'numTraining': 221, 'file_ending': '.mha'}, 'unpack_dataset': True, 'device': device(type='cuda')}",
-    "network": "ResidualEncoderUNet",
-    "num_epochs": "1000",
-    "num_input_channels": "1",
-    "num_iterations_per_epoch": "250",
-    "num_val_iterations_per_epoch": "50",
-    "optimizer": "SGD (\nParameter Group 0\n    dampening: 0\n    differentiable: False\n    foreach: None\n    fused: None\n    initial_lr: 0.01\n    lr: 0.01\n    maximize: False\n    momentum: 0.99\n    nesterov: True\n    weight_decay: 3e-05\n)",
-    "output_folder": "/datasets/work/hb-synthrad2023/work/synthrad2025/bw_workplace/data/nnunet_struct/results/Dataset262_synthrad2025_task1_MR_HN_pre_v2r_stitched_masked/nnUNetTrainerMRCT_loss_masked_perception_masked__nnUNetResEncUNetLPlans__3d_fullres/fold_4",
-    "output_folder_base": "/datasets/work/hb-synthrad2023/work/synthrad2025/bw_workplace/data/nnunet_struct/results/Dataset262_synthrad2025_task1_MR_HN_pre_v2r_stitched_masked/nnUNetTrainerMRCT_loss_masked_perception_masked__nnUNetResEncUNetLPlans__3d_fullres",
-    "oversample_foreground_percent": "0.33",
-    "perception_masked": "True",
-    "plans_manager": "{'dataset_name': 'Dataset262_synthrad2025_task1_MR_HN_pre_v2r_stitched_masked', 'plans_name': 'nnUNetResEncUNetLPlans', 'original_median_spacing_after_transp': [3.0, 1.0, 1.0], 'original_median_shape_after_transp': [89, 296, 279], 'image_reader_writer': 'SimpleITKIO', 'transpose_forward': [0, 1, 2], 'transpose_backward': [0, 1, 2], 'configurations': {'2d': {'data_identifier': 'nnUNetPlans_2d', 'preprocessor_name': 'DefaultPreprocessor', 'batch_size': 30, 'patch_size': [320, 320], 'median_image_size_in_voxels': [296.0, 279.0], 'spacing': [1.0, 1.0], 'normalization_schemes': ['ZScoreNormalization'], 'use_mask_for_norm': [False], 'resampling_fn_data': 'resample_data_or_seg_to_shape', 'resampling_fn_seg': 'resample_data_or_seg_to_shape', 'resampling_fn_data_kwargs': {'is_seg': False, 'order': 3, 'order_z': 0, 'force_separate_z': None}, 'resampling_fn_seg_kwargs': {'is_seg': True, 'order': 1, 'order_z': 0, 'force_separate_z': None}, 'resampling_fn_probabilities': 'resample_data_or_seg_to_shape', 'resampling_fn_probabilities_kwargs': {'is_seg': False, 'order': 1, 'order_z': 0, 'force_separate_z': None}, 'architecture': {'network_class_name': 'dynamic_network_architectures.architectures.unet.ResidualEncoderUNet', 'arch_kwargs': {'n_stages': 7, 'features_per_stage': [32, 64, 128, 256, 512, 512, 512], 'conv_op': 'torch.nn.modules.conv.Conv2d', 'kernel_sizes': [[3, 3], [3, 3], [3, 3], [3, 3], [3, 3], [3, 3], [3, 3]], 'strides': [[1, 1], [2, 2], [2, 2], [2, 2], [2, 2], [2, 2], [2, 2]], 'n_blocks_per_stage': [1, 3, 4, 6, 6, 6, 6], 'n_conv_per_stage_decoder': [1, 1, 1, 1, 1, 1], 'conv_bias': True, 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm2d', 'norm_op_kwargs': {'eps': 1e-05, 'affine': True}, 'dropout_op': None, 'dropout_op_kwargs': None, 'nonlin': 'torch.nn.LeakyReLU', 'nonlin_kwargs': {'inplace': True}}, '_kw_requires_import': ['conv_op', 'norm_op', 'dropout_op', 'nonlin']}, 'batch_dice': True}, '3d_fullres': {'data_identifier': 'nnUNetPlans_3d_fullres', 'preprocessor_name': 'DefaultPreprocessor', 'batch_size': 2, 'patch_size': [56, 192, 160], 'median_image_size_in_voxels': [89.0, 296.0, 279.0], 'spacing': [3.0, 1.0, 1.0], 'normalization_schemes': ['ZScoreNormalization'], 'use_mask_for_norm': [False], 'resampling_fn_data': 'resample_data_or_seg_to_shape', 'resampling_fn_seg': 'resample_data_or_seg_to_shape', 'resampling_fn_data_kwargs': {'is_seg': False, 'order': 3, 'order_z': 0, 'force_separate_z': None}, 'resampling_fn_seg_kwargs': {'is_seg': True, 'order': 1, 'order_z': 0, 'force_separate_z': None}, 'resampling_fn_probabilities': 'resample_data_or_seg_to_shape', 'resampling_fn_probabilities_kwargs': {'is_seg': False, 'order': 1, 'order_z': 0, 'force_separate_z': None}, 'architecture': {'network_class_name': 'dynamic_network_architectures.architectures.unet.ResidualEncoderUNet', 'arch_kwargs': {'n_stages': 6, 'features_per_stage': [32, 64, 128, 256, 320, 320], 'conv_op': 'torch.nn.modules.conv.Conv3d', 'kernel_sizes': [[1, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3], [3, 3, 3]], 'strides': [[1, 1, 1], [1, 2, 2], [2, 2, 2], [2, 2, 2], [2, 2, 2], [1, 2, 2]], 'n_blocks_per_stage': [1, 3, 4, 6, 6, 6], 'n_conv_per_stage_decoder': [1, 1, 1, 1, 1], 'conv_bias': True, 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm3d', 'norm_op_kwargs': {'eps': 1e-05, 'affine': True}, 'dropout_op': None, 'dropout_op_kwargs': None, 'nonlin': 'torch.nn.LeakyReLU', 'nonlin_kwargs': {'inplace': True}}, '_kw_requires_import': ['conv_op', 'norm_op', 'dropout_op', 'nonlin']}, 'batch_dice': False}}, 'experiment_planner_used': 'nnUNetPlannerResEncL', 'label_manager': 'LabelManager', 'foreground_intensity_properties_per_channel': {'0': {'max': 4142.0, 'mean': 75.25260162353516, 'median': 0.0, 'min': 0.0, 'percentile_00_5': 0.0, 'percentile_99_5': 1222.0, 'std': 197.03660583496094}}}",
-    "preprocessed_dataset_folder": "/datasets/work/hb-synthrad2023/work/synthrad2025/bw_workplace/data/nnunet_struct/preprocessed/Dataset262_synthrad2025_task1_MR_HN_pre_v2r_stitched_masked/nnUNetPlans_3d_fullres",
-    "preprocessed_dataset_folder_base": "/datasets/work/hb-synthrad2023/work/synthrad2025/bw_workplace/data/nnunet_struct/preprocessed/Dataset262_synthrad2025_task1_MR_HN_pre_v2r_stitched_masked",
-    "save_every": "50",
-    "torch_version": "2.7.1+cu126",
-    "unpack_dataset": "True",
-    "was_initialized": "True",
-    "weight_decay": "3e-05"
-}

Dataset262/nnUNetTrainerMRCT_loss_masked_perception_masked__nnUNetResEncUNetLPlans__3d_fullres/fold_4/validation/summary.json

@@ -1,52 +0,0 @@
-{
-    "mae": {
-        "mean": 79.01048282643166,
-        "std": 25.814640283626616,
-        "max": 136.18736974287893,
-        "min": 48.097312855892966,
-        "25pc": 56.43015459462086,
-        "50pc": 70.26204491277278,
-        "75pc": 100.98830817898656,
-        "count": 44
-    },
-    "psnr": {
-        "mean": 28.16481870353614,
-        "std": 2.7080036499500273,
-        "max": 31.785287333787497,
-        "min": 23.128256951315468,
-        "25pc": 25.650981331586912,
-        "50pc": 28.757230793948047,
-        "75pc": 30.502033048682343,
-        "count": 44
-    },
-    "ms_ssim": {
-        "mean": 0.9456753494688663,
-        "std": 0.03748245085540397,
-        "max": 0.9847741314776761,
-        "min": 0.8537015800968059,
-        "25pc": 0.9215156936718175,
-        "50pc": 0.9640495073026166,
-        "75pc": 0.9752980815606082,
-        "count": 44
-    },
-    "DICE": {
-        "mean": 0.8262107798999007,
-        "std": 0.11717953033145945,
-        "max": 0.9291353821754456,
-        "min": 0.43098184466362,
-        "25pc": 0.7794158011674881,
-        "50pc": 0.8882485926151276,
-        "75pc": 0.906611979007721,
-        "count": 44
-    },
-    "HD95": {
-        "mean": 2.321003418077122,
-        "std": 1.4156960699276717,
-        "max": 7.2080817222595215,
-        "min": 1.0942808389663696,
-        "25pc": 1.3163809180259705,
-        "50pc": 1.8190175890922546,
-        "75pc": 2.7259116172790527,
-        "count": 44
-    }
-}