Closed woohee-yang closed 8 months ago
dataset: MaskBaseDataset augmentation: AutoAugmentation resize: [ 256, 192 ] valid_ratio: 0.2 balanced_split: true sampler: ImbalancedSampler age_drop: True
seed: 42 max_epochs: 30 batch_size: 48 valid_batch_size: 64 optimizer: AdamW lr: 0.001 criterion: focal scheduler: cosine lr_decay_step: 20 log_interval: 50
config
==== Dataset & Transform ====
dataset: MaskBaseDataset augmentation: AutoAugmentation resize: [ 256, 192 ] valid_ratio: 0.2 balanced_split: true sampler: ImbalancedSampler age_drop: True
==== Training ====
seed: 42 max_epochs: 30 batch_size: 48 valid_batch_size: 64 optimizer: AdamW lr: 0.001 criterion: focal scheduler: cosine lr_decay_step: 20 log_interval: 50
inference score