_meta_: {} bundle_root: /Users/sakshirathi/Downloads/work_dir/segresnet_0 ckpt_path: $@bundle_root + '/model' mlflow_tracking_uri: $@ckpt_path + '/mlruns/' mlflow_experiment_name: Auto3DSeg data_file_base_dir: /Users/sakshirathi/Documents/ShamLab data_list_file_path: /Users/sakshirathi/Downloads/work_dir/dataset_local.json modality: ct fold: 0 input_channels: 1 output_classes: 2 class_names: null class_index: null debug: false ckpt_save: true cache_rate: null roi_size: [384, 384, 60] auto_scale_allowed: true auto_scale_batch: true auto_scale_roi: false auto_scale_filters: false quick: false channels_last: true validate_final_original_res: true calc_val_loss: false amp: true log_output_file: null cache_class_indices: null early_stopping_fraction: 0.001 determ: false orientation_ras: true crop_foreground: true learning_rate: 0.0002 batch_size: 1 num_images_per_batch: 1 num_epochs: 1250 num_warmup_epochs: 3 sigmoid: false resample: true resample_resolution: [0.48766356436698155, 0.4876635832539761, 2.748479210553717] crop_mode: ratio normalize_mode: range intensity_bounds: [39.63595217750186, 97.59593563988095] num_epochs_per_validation: null num_epochs_per_saving: 1 num_workers: 4 num_steps_per_image: null num_crops_per_image: 2 loss: {_target_: DiceCELoss, include_background: true, squared_pred: true, smooth_nr: 0, smooth_dr: 1.0e-05, softmax: $not @sigmoid, sigmoid: $@sigmoid, to_onehot_y: $not @sigmoid} optimizer: {_target_: torch.optim.AdamW, lr: '@learning_rate', weight_decay: 1.0e-05} network: _target_: SegResNetDS init_filters: 32 blocks_down: [1, 2, 2, 4, 4] norm: INSTANCE_NVFUSER in_channels: '@input_channels' out_channels: '@output_classes' dsdepth: 4 finetune: {enabled: false, ckpt_name: $@bundle_root + '/model/model.pt'} validate: {enabled: false, ckpt_name: $@bundle_root + '/model/model.pt', output_path: $@bundle_root + '/prediction_validation', save_mask: false, invert: true} infer: {enabled: false, ckpt_name: $@bundle_root + '/model/model.pt', output_path: $@bundle_root + '/prediction_' + @infer#data_list_key, data_list_key: testing} anisotropic_scales: true spacing_median: [0.48766356436698155, 0.4876635832539761, 4.770811902267695] spacing_lower: [0.42813486948609353, 0.428134856247896, 2.499999978382533] spacing_upper: [0.5859375, 0.5859375004856939, 5.012642938162783] image_size_mm_median: [249.68374495589455, 249.68375462603575, 168.30083390623668] image_size_mm_90: [265.61599121093747, 265.6159922216141, 190.12765338720757] image_size: [544, 544, 69]