File size: 54,945 Bytes
c0c4fc6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | {
"_best_ema": "None",
"_best_ema_f1": "None",
"alpha": "0",
"batch_size": "141",
"beta": "1",
"configuration_manager": "{'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 6, 'features_per_stage': [64, 128, 256, 512, 1024, 2048], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[5], [5], [5], [5], [5], [5]], 'strides': [[1], [2], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}",
"configuration_name": "1d_decoding",
"cudnn_version": 90100,
"current_epoch": "0",
"dataloader_train": "<batchgenerators.dataloading.nondet_multi_threaded_augmenter.NonDetMultiThreadedAugmenter object at 0x7f8ca9435840>",
"dataloader_train.generator": "<nnunetv2.training.dataloading.data_loader_1d_classification.nnUNetWithClassificationDataLoader1D object at 0x7f8ca9437400>",
"dataloader_train.num_processes": "12",
"dataloader_train.transform": "None",
"dataloader_val": "<batchgenerators.dataloading.nondet_multi_threaded_augmenter.NonDetMultiThreadedAugmenter object at 0x7f8ca9435cf0>",
"dataloader_val.generator": "<nnunetv2.training.dataloading.data_loader_1d_classification.nnUNetWithClassificationDataLoader1D object at 0x7f8ca9437340>",
"dataloader_val.num_processes": "6",
"dataloader_val.transform": "None",
"dataset_json": "{'channel_names': {'0': 'LeadII', '1': 'LeadV1', '2': 'LeadV6'}, 'labels': {'background': 0, 'p_wave': 1, 'qrs_wave': 2, 't_wave': 3, 'noise': 4, 'qrs_abnormal': 5}, 'use_for_validation': {'background': False, 'p_wave': True, 'qrs_wave': True, 't_wave': True, 'noise': True, 'qrs_abnormal': True}, 'classification_head': True, 'binary': True, 'numTraining': 87800, 'file_ending': '.npy'}",
"device": "cuda:0",
"disable_checkpointing": "False",
"enable_deep_supervision": "True",
"fold": "2",
"folder_with_segs_from_previous_stage": "None",
"gpu_name": "NVIDIA GeForce RTX 3090",
"grad_scaler": "<torch.cuda.amp.grad_scaler.GradScaler object at 0x7f8cbc302ad0>",
"hostname": "betal112465",
"inference_allowed_mirroring_axes": "(0,)",
"initial_lr": "0.0005",
"is_cascaded": "False",
"is_ddp": "False",
"label_manager": "<nnunetv2.utilities.label_handling.label_handling.LabelManager object at 0x7f8cbc303190>",
"local_rank": "0",
"log_file": "/home/lukas/UU/ASRA/ALADINv2/data/nnUNet_results/Dataset201_all_101/ClassificationTrainer__nnUNetWithClassificationPlans__1d_decoding/fold_2/training_log_2025_12_2_07_02_28.txt",
"logger": "<nnunetv2.training.logging.nnunet_logger.nnUNetWithClassificationLogger object at 0x7f8cbc59d510>",
"loss": "(DeepSupervisionWrapper(\n (loss): DC_and_BCE_loss(\n (ce): BCEWithLogitsLoss()\n (dc): OptimizedModule(\n (_orig_mod): MemoryEfficientSoftDiceLoss()\n )\n )\n), CrossEntropyLoss())",
"lr_scheduler": "<nnunetv2.training.lr_scheduler.polylr.PolyLRScheduler object at 0x7f8cbc10e770>",
"my_init_kwargs": "{'plans': {'dataset_name': 'Dataset201_all_101', 'plans_name': 'nnUNetWithClassificationPlans', 'original_median_spacing_after_transp': [999.0, 999.0, 1.0], 'original_median_shape_after_transp': [1, 1, 2048], 'image_reader_writer': 'NumpyIO', 'transpose_forward': [0, 1, 2], 'transpose_backward': [0, 1, 2], 'configurations': {'1d_encoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 6, 'features_per_stage': [64, 128, 256, 512, 1024, 2048], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[5], [5], [5], [5], [5], [5]], 'strides': [[1], [2], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2, 2], 'conv_bias': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 6, 'features_per_stage': [64, 128, 256, 512, 1024, 2048], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[5], [5], [5], [5], [5], [5]], 'strides': [[1], [2], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_2': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 1024, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 6, 'features_per_stage': [64, 128, 256, 512, 1024, 2048], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[7], [7], [7], [7], [7], [7]], 'strides': [[1], [2], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2, 2], 'conv_bias': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512, 256, 128], [512, 256, 128]], 'kernel_sizes_classification_branch': [[7, 7, 7], [7, 7, 7]], 'strides_classification_branch': [[2, 2, 2], [2, 2, 2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_small_encoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [64, 128, 256, 512, 1024], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[7], [7], [7], [7], [7]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': True, 'use_decoding_layers_classification_branch': False, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_small_encoding_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [64, 128, 256, 512, 1024], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[7], [7], [7], [7], [7]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': True, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_small_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [64, 128, 256, 512, 1024], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[7], [7], [7], [7], [7]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_encoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[9], [9], [9], [9], [9]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_encoding_2': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[11], [11], [11], [11], [11]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': True, 'use_decoding_layers_classification_branch': False, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512, 256, 128], [512, 256, 128]], 'kernel_sizes_classification_branch': [[17, 17, 17], [17, 17, 17]], 'strides_classification_branch': [[2, 2, 2], [2, 2, 2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_decoding_2': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[11], [11], [11], [11], [11]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512, 256, 128], [512, 256, 128]], 'kernel_sizes_classification_branch': [[17, 17, 17], [17, 17, 17]], 'strides_classification_branch': [[2, 2, 2], [2, 2, 2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_3p_encoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNet3pWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[9], [9], [9], [9], [9]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': True, 'use_decoding_layers_classification_branch': False, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_3p_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNet3pWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[9], [9], [9], [9], [9]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[9], [9], [9], [9], [9]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_encoding_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[9], [9], [9], [9], [9]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': True, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}}, 'experiment_planner_used': 'UNetWithClassificationPlanner', 'label_manager': 'LabelManager', 'foreground_intensity_properties_per_channel': {'0': {'max': 10.486477851867676, 'mean': -0.19572360422609777, 'median': -0.20229651033878326, 'min': -14.818507194519043, 'percentile_00_5': -2.0146844387054443, 'percentile_99_5': 1.8262673616409302, 'std': 0.5616268453683911}}}, 'configuration': '1d_decoding', 'fold': 2, 'dataset_json': {'channel_names': {'0': 'LeadII', '1': 'LeadV1', '2': 'LeadV6'}, 'labels': {'background': 0, 'p_wave': 1, 'qrs_wave': 2, 't_wave': 3, 'noise': 4, 'qrs_abnormal': 5}, 'use_for_validation': {'background': False, 'p_wave': True, 'qrs_wave': True, 't_wave': True, 'noise': True, 'qrs_abnormal': True}, 'classification_head': True, 'binary': True, 'numTraining': 87800, 'file_ending': '.npy'}, 'unpack_dataset': True, 'device': device(type='cuda')}",
"network": "UNetWithClassificationBranches",
"num_epochs": "32",
"num_input_channels": "3",
"num_iterations_per_epoch": "100",
"num_val_iterations_per_epoch": "20",
"optimizer": "AdamW (\nParameter Group 0\n amsgrad: True\n betas: (0.9, 0.999)\n capturable: False\n differentiable: False\n eps: 1e-08\n foreach: None\n fused: None\n initial_lr: 0.0005\n lr: 0.0005\n maximize: False\n weight_decay: 3e-05\n)",
"output_folder": "/home/lukas/UU/ASRA/ALADINv2/data/nnUNet_results/Dataset201_all_101/ClassificationTrainer__nnUNetWithClassificationPlans__1d_decoding/fold_2",
"output_folder_base": "/home/lukas/UU/ASRA/ALADINv2/data/nnUNet_results/Dataset201_all_101/ClassificationTrainer__nnUNetWithClassificationPlans__1d_decoding",
"oversample_foreground_percent": "0",
"plans_manager": "{'dataset_name': 'Dataset201_all_101', 'plans_name': 'nnUNetWithClassificationPlans', 'original_median_spacing_after_transp': [999.0, 999.0, 1.0], 'original_median_shape_after_transp': [1, 1, 2048], 'image_reader_writer': 'NumpyIO', 'transpose_forward': [0, 1, 2], 'transpose_backward': [0, 1, 2], 'configurations': {'1d_encoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 6, 'features_per_stage': [64, 128, 256, 512, 1024, 2048], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[5], [5], [5], [5], [5], [5]], 'strides': [[1], [2], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2, 2], 'conv_bias': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 6, 'features_per_stage': [64, 128, 256, 512, 1024, 2048], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[5], [5], [5], [5], [5], [5]], 'strides': [[1], [2], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_2': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 1024, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 6, 'features_per_stage': [64, 128, 256, 512, 1024, 2048], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[7], [7], [7], [7], [7], [7]], 'strides': [[1], [2], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2, 2], 'conv_bias': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512, 256, 128], [512, 256, 128]], 'kernel_sizes_classification_branch': [[7, 7, 7], [7, 7, 7]], 'strides_classification_branch': [[2, 2, 2], [2, 2, 2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_small_encoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [64, 128, 256, 512, 1024], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[7], [7], [7], [7], [7]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': True, 'use_decoding_layers_classification_branch': False, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_small_encoding_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [64, 128, 256, 512, 1024], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[7], [7], [7], [7], [7]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': True, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_small_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [64, 128, 256, 512, 1024], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[7], [7], [7], [7], [7]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_encoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[9], [9], [9], [9], [9]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_encoding_2': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[11], [11], [11], [11], [11]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': True, 'use_decoding_layers_classification_branch': False, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512, 256, 128], [512, 256, 128]], 'kernel_sizes_classification_branch': [[17, 17, 17], [17, 17, 17]], 'strides_classification_branch': [[2, 2, 2], [2, 2, 2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_decoding_2': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[11], [11], [11], [11], [11]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512, 256, 128], [512, 256, 128]], 'kernel_sizes_classification_branch': [[17, 17, 17], [17, 17, 17]], 'strides_classification_branch': [[2, 2, 2], [2, 2, 2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_3p_encoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNet3pWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[9], [9], [9], [9], [9]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': True, 'use_decoding_layers_classification_branch': False, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_3p_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNet3pWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[9], [9], [9], [9], [9]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[9], [9], [9], [9], [9]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': False, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}, '1d_tiny_encoding_decoding': {'data_identifier': 'nnUNetWithClassificationPlans_1d', 'preprocessor_name': 'WithClassificationPreprocessor', 'batch_size': 141, 'patch_size': [2048], 'median_image_size_in_voxels': [2048.0], 'spacing': [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': 'nnunetv2.models.custom_architectures.UNetWithClassificationBranches', 'arch_kwargs': {'n_stages': 5, 'features_per_stage': [32, 64, 128, 256, 512], 'conv_op': 'torch.nn.modules.conv.Conv1d', 'kernel_sizes': [[9], [9], [9], [9], [9]], 'strides': [[1], [2], [2], [2], [2]], 'n_conv_per_stage': [2, 2, 2, 2, 2], 'n_conv_per_stage_decoder': [2, 2, 2, 2], 'conv_bias': True, 'use_encoding_layers_classification_branch': True, 'use_decoding_layers_classification_branch': True, 'num_classes_classification_branch': [2, 2], 'features_per_stage_classification_branch': [[512], [512]], 'kernel_sizes_classification_branch': [[17], [17]], 'strides_classification_branch': [[2], [2]], 'norm_op': 'torch.nn.modules.instancenorm.InstanceNorm1d', '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}}, 'experiment_planner_used': 'UNetWithClassificationPlanner', 'label_manager': 'LabelManager', 'foreground_intensity_properties_per_channel': {'0': {'max': 10.486477851867676, 'mean': -0.19572360422609777, 'median': -0.20229651033878326, 'min': -14.818507194519043, 'percentile_00_5': -2.0146844387054443, 'percentile_99_5': 1.8262673616409302, 'std': 0.5616268453683911}}}",
"preprocessed_dataset_folder": "/home/lukas/UU/ASRA/ALADINv2/data/nnUNet_preprocessed/Dataset201_all_101/nnUNetWithClassificationPlans_1d",
"preprocessed_dataset_folder_base": "/home/lukas/UU/ASRA/ALADINv2/data/nnUNet_preprocessed/Dataset201_all_101",
"save_every": "50",
"torch_version": "2.5.1+cu124",
"unpack_dataset": "True",
"was_initialized": "True",
"weight_decay": "3e-05"
} |