| seed: 1337 | |
| output_dir: outputs/default | |
| amp: true | |
| compile: false | |
| channels_last_3d: false | |
| num_workers: 4 | |
| pin_memory: true | |
| wandb: | |
| enabled: true | |
| project: SACFlow-FM | |
| entity: null | |
| mode: online | |
| tags: [] | |
| log_images_every: 250 | |
| watch_model: false | |
| distributed: | |
| backend: nccl | |
| find_unused_parameters: false | |
| data: | |
| manifest: data_json/mmwhs_ct2mr.json | |
| spacing: null | |
| patch_size: [96, 96, 96] | |
| batch_size: 1 | |
| val_batch_size: 1 | |
| cache_rate: 0.0 | |
| in_channels: 1 | |
| num_classes: 5 | |
| foreground_labels: null | |
| intensity: | |
| clip: null | |
| normalize: zscore_nonzero | |
| augmentation: | |
| random_flip: true | |
| random_intensity_shift: 0.10 | |
| random_intensity_scale: 0.10 | |
| model: | |
| name: unet3d | |
| in_channels: 1 | |
| num_classes: 5 | |
| base_channels: 32 | |
| levels: 4 | |
| norm: instance | |
| dropout: 0.0 | |
| deep_supervision: false | |
| adapter: | |
| enabled: false | |
| bottleneck_ratio: 0.25 | |
| locations: [prelogit] | |
| optim: | |
| optimizer: adamw | |
| lr: 2.0e-4 | |
| weight_decay: 1.0e-4 | |
| betas: [0.9, 0.999] | |
| scheduler: cosine | |
| warmup_steps: 500 | |
| max_steps: 30000 | |
| grad_clip_norm: 12.0 | |
| train: | |
| mode: source_train | |
| epochs: 100 | |
| steps_per_epoch: 250 | |
| val_every: 1 | |
| save_every: 5 | |
| log_every: 20 | |
| resume: null | |
| source_checkpoint: null | |
| memory_path: null | |
| ema_decay: 0.99 | |
| pseudo_conf_threshold: 0.70 | |
| class_balanced_thresholds: false | |
| loss: | |
| ce: 1.0 | |
| dice: 1.0 | |
| entropy: 0.0 | |
| consistency: 0.0 | |
| eval: | |
| checkpoint: null | |
| split: target_test | |
| save_predictions: false | |
| sliding_window: true | |
| roi_size: [96, 96, 96] | |
| sw_batch_size: 1 | |
| overlap: 0.5 | |