| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102 |
- trainer:
- name: supervised_segmentation
- train:
- seed: 42
- epochs: 200
- batch_size: 4
- val_batch_size: 4
- amp: true
- num_workers: 4
- pin_memory: true
- persistent_workers: true
- prefetch_factor: 2
- device: cuda
- grad_clip:
- enabled: true
- max_norm: 1.0
- norm_type: 2.0
- auto_batch_size:
- enabled: false
- target_memory_fraction: 0.75
- reference_gpu_gb: 8.0
- reference_batch_size: 4
- min_batch_size: 1
- max_batch_size: 8
- metrics:
- task_mode: binary
- metrics:
- - name: dice
- - name: iou
- validation:
- enabled: true
- interval: 1
- threshold: 0.5
- early_stopping: true
- early_stopping_patience: 40
- early_stopping_min_delta: 0.0
- metrics:
- task_mode: binary
- metrics:
- - name: dice
- - name: iou
- dataset:
- name: ultrasound_sup_seg
- dataset_name: BUSI
- root: data/BUSI
- split: train
- split_file: null
- val_split: val
- val_split_file: null
- image_size: [256, 256]
- in_channels: 3
- num_classes: 1
- model:
- model_name: swinv2_tiny_patch4_window8_256
- load_weights: false
- decoder_channels: [384, 192, 96, 96]
- fwta_wavelet: haar
- fwta_level: 1
- fwta_sigma_ratio: 0.35
- fwta_tau_fourier: 0.15
- fwta_gate_temperature: 1.0
- use_multiscale_features: true
- include_patch_embed: true
- optimizer:
- name: adamw
- lr: 1.0e-4
- weight_decay: 0.05
- scheduler:
- name: cosine
- warmup:
- name: linear
- params:
- start_factor: 0.1
- total_iters: 10
- params:
- T_max: 190
- eta_min: 1.0e-6
- checkpoint:
- dir: outputs/supervised_segmentation/train_sup_us_template
- save: true
- save_last: true
- monitor: dice
- monitor_mode: max
- resume: null
- resume_strict: true
- resume_training: true
- logging:
- log_interval: 10
- print_training_setup: true
- use_swanlab: true
- project: X_SSL_Net
- experiment_name: supervised_segmentation
- swanlab_mode: null
|
