| 1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192 |
- 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
- 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:
- dataset_name: BUSI
- root: data/BUSI
- split: train
- val_split: val
- 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/experiments/supervised/BUSI
- 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: sup_busi
- swanlab_mode: null
|
