| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118 |
- trainer:
- name: supervised_segmentation
- train:
- seed: 42
- deterministic: false
- epochs: 200
- batch_size: 4
- val_batch_size: 4
- accum_steps: 1
- 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
- loss:
- name: dicece
- task_mode: binary
- params:
- include_background: true
- lambda_dice: 0.7
- lambda_ce: 0.3
- 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]
- 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
- augmentation:
- train:
- random_flip: true
- random_rotate_90: true
- random_brightness_contrast: true
- brightness_limit: 0.15
- contrast_limit: 0.15
- random_gaussian_noise: true
- gaussian_noise_std: 0.03
- val: {}
- 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
|
