config.json

{
    "use_wandb": true,
    "use_cm": false,
    "entity":"connect-cv-13_2",
    "project_name": "Sementic-Segmentation-HJH",
    "exp_name": "exp", 
    "exp_num": 2    ,
    "user_name": "jh", 
    "datadir": "../input/data/",
    "train_path": "train/DCM/",
    "label_path": "train/outputs_json/",
    "valid_file": "valid.csv",
    "test_file": "test.csv",
    "transform": ["resize"],
    "seed": 13,
    "model_name": "SegModel",
    "model_param":
    {
        "pretrained": true,
        "backbone": "resnet18"
    },
    "classes": [
        "finger-1", "finger-2", "finger-3", "finger-4", "finger-5",
        "finger-6", "finger-7", "finger-8", "finger-9", "finger-10",
        "finger-11", "finger-12", "finger-13", "finger-14", "finger-15",
        "finger-16", "finger-17", "finger-18", "finger-19", "Trapezium",
        "Trapezoid", "Capitate", "Hamate", "Scaphoid", "Lunate",
        "Triquetrum", "Pisiform", "Radius", "Ulna"
    ],
    "num_classes": 29,
    "batch_size": 8,
    "opt_name": "Adam",
    "loss": "bce_with_logit_loss",
    "loss_param":
    {
        
    },
    "lr": 1e-4,
    "lr_scheduler":"",
    "lr_scheduler_param":
    {
        "step_size":3
    },
    "weight_decay": 1e-6,
    "epochs": 150,
    "savedir": "./checkpoint",
    "grad_accum_steps": 1,
    "mixed_precision": "fp16",
    "patience": 100
}