| _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] |
|
|
