base_config.json

{
    "name": "Baseline",
    "root_dir": "/opt/ml/data1024",
    "seed": 21,
    "epochs": 250,
    "early_stop": 10,
    "save_interval": 1,
    "log_interval": 20,
    "is_wandb": false,
    "is_debug": true,
    "smp": {
        "use": true,
        "args": {
            "encoder_name": "efficientnet-b7",
            "encoder_weights": "imagenet",
            "in_channels": 3,
            "classes": 29
        }
    },
    "dataset": "XRayDataset",
    "augmentation": "BaseAugmentation",
    "batch_size": 8,
    "iters_to_accumulate": 1,
    "model": "Unet",
    "criterion": [
        "combo_loss"
    ],
    "optimizer": {
        "type": "AdamW",
        "args": {
            "lr": 1e-4,
            "weight_decay": 1e-5
        }
    },
    "lr_scheduler": {
        "type": "CosineAnnealingWarmupRestarts",
        "args": {
            "first_cycle_steps": 1000,
            "cycle_mult": 2.0,
            "max_lr": 1e-4,
            "min_lr": 1e-5,
            "warmup_steps": 50,
            "gamma": 0.8
        }
    },
    "num_workers": 4
}