train.json

{
    "data_dir": {
        "chexpert": "data/classification/chest/cheXpert/CheXpert-v1.0-small/",
        "nihchest": "data/classification/chest/nih/",
        "mimic": "data/classification/chest/mimic-cxr/",
        "all": "data/classification/chest/all/",
    },
    "model_list": [
      "resnet50", "resnet101", "resnet152",
      "convnext_tiny", "convnext_small", "convnext_base",
      "densenet121", "densenet161", "densenet169", "densenet201",
      "deit_tiny_patch16_224", "deit_small_patch16_224", "deit_base_patch16_224",
      "swin_tiny_patch4_window7_224", "swin_small_patch4_window7_224", "swin_base_patch4_window7_224"
    ],
    "model_weight": {
      "resnet50_ADNet_IN1K": "pretrained/resnet50_ADNet_IN1K.pth.tar",
      "resnet50_ADNet_ALL": "pretrained/resnet50_ADNet_ALL.pth.tar",
      "convnext_tiny_ADNet_IN1K": "pretrained/convnext_tiny_ADNet_IN1K",
      "convnext_tiny_ADNet_ALL": "pretrained/convnext_tiny_ADNet_ALL"
    },
    "settings": {
      "chexpert": {
        "dataset_type": "chexpert", "pretrained": true, "entity": "hankyul2",
        "train_size": [256, 256], "test_size": [256, 256], "center_crop_ptr": 0.95, "hflip": 0.5,
        "remode": 0.2, "recount": 3, "lr": 5e-5, "min_lr": 5e-6, "auto_aug": "rand-m9-mstd0.9",
        "batch_size": 32, "epoch": 20, "optimizer": "adamw", "weight_decay": 1e-4,
        "scheduler": "cosine", "warmup_epoch": 0, "warmup_lr": 1e-6, "criterion": "bce", "patient_epoch": 3,
        "save_weight": true, "print_freq": 100, "drop_path_rate": 0.05, "valid_freq": 200
      },
      "nihchest": {
        "dataset_type": "nihchest", "pretrained": true, "entity": "hankyul2", "use_wandb": true,
        "train_size": [352, 352], "test_size": [448, 448], "center_crop_ptr": 0.95, "hflip": 0.5,
        "remode": 0.2, "recount": 3, "lr": 1e-4, "min_lr": 1e-6, "auto_aug": "rand-m9-mstd0.5",
        "batch_size": 30, "epoch": 30, "optimizer": "adamw", "weight_decay": 1e-4,
        "scheduler": "cosine", "warmup_epoch": 0, "warmup_lr": 1e-6, "criterion": "bce", "patient_epoch": 3,
        "save_weight": true, "print_freq": 100, "drop_path_rate": 0.05, "ema": true, "ema_decay": 0.99995, "aug_repeat": 3
      },
      "all": {
        "dataset_type": "all", "pretrained": true, "entity": "hankyul2", "use_wandb": true,
        "train_size": [352, 352], "test_size": [416, 416], "center_crop_ptr": 0.95, "hflip": 0.5,
        "remode": 0.2, "recount": 3, "lr": 4e-4, "min_lr": 1e-6, "auto_aug": "rand-m9-mstd0.5",
        "batch_size": 30, "epoch": 30, "optimizer": "adamw", "weight_decay": 1e-4,
        "scheduler": "cosine", "warmup_epoch": 0, "warmup_lr": 1e-6, "criterion": "bce", "patient_epoch": 3,
        "save_weight": true, "print_freq": 100, "drop_path_rate": 0.05, "ema": true
      },
      "mimic": {
        "dataset_type": "mimic", "pretrained": true, "entity": "hankyul2", "use_wandb": true,
        "train_size": [256, 256], "test_size": [256, 256], "center_crop_ptr": 0.95, "hflip": 0.5,
        "remode": 0.2, "recount": 3, "lr": 1e-4, "min_lr": 1e-6, "auto_aug":"rand-m9-mstd0.5", "ema": true,
        "batch_size": 32, "epoch": 30, "optimizer": "adamw", "weight_decay": 1e-4,
        "scheduler": "cosine", "warmup_epoch": 0, "warmup_lr": 1e-6, "criterion": "bce", "patient_epoch": 3,
        "save_weight": true, "print_freq": 100, "drop_path_rate": 0.05
      }
  }
}