{
  "seed": 42,
  "data": {
    "dataset_name": "HlexNC/chest-xray-14-320",
    "image_size": 320,
    "num_workers": 4,
    "pin_memory": true,
    "train_split": 0.7,
    "val_split": 0.1,
    "test_split": 0.2,
    "augmentation": {
      "horizontal_flip": true,
      "rotation_degrees": 15,
      "affine_translate": 0.05,
      "affine_shear": 5,
      "color_jitter": {
        "brightness": 0.3,
        "contrast": 0.3,
        "saturation": 0.05
      },
      "gaussian_blur_prob": 0.2,
      "random_erasing_prob": 0.1
    },
    "labels": [
      "Atelectasis",
      "Cardiomegaly",
      "Effusion",
      "Infiltration",
      "Mass",
      "Nodule",
      "Pneumonia",
      "Pneumothorax",
      "Consolidation",
      "Edema",
      "Emphysema",
      "Fibrosis",
      "Pleural_Thickening",
      "Hernia"
    ],
    "data_dir": "/kaggle/working/data",
    "hf_dataset_repo": "HlexNC/chest-xray-14-320",
    "hf_dataset_revision": "44443e6ee968b3c6094b63f14a27698c40b50680"
  },
  "training": {
    "epochs": 100,
    "batch_size": 24,
    "learning_rate": 0.001,
    "weight_decay": 0.0001,
    "optimizer": "adamw",
    "scheduler": "cosine",
    "warmup_epochs": 5,
    "early_stopping_patience": 15,
    "gradient_clip_norm": 1.0,
    "use_amp": true,
    "grad_accum_steps": 4,
    "multilabel_weight": 1.0,
    "binary_weight": 0.5
  },
  "evaluation": {
    "primary_metric": "auc_roc_macro",
    "threshold": 0.5
  },
  "logging": {
    "log_dir": "logs/",
    "checkpoint_dir": "/kaggle/working/checkpoints",
    "save_every_n_epochs": 5,
    "log_every_n_steps": 100
  },
  "huggingface": {
    "dataset_repo": "HlexNC/chest-xray-14-320",
    "scratch_model_repo": "HlexNC/chexvision-scratch",
    "transfer_model_repo": "HlexNC/chexvision-densenet",
    "space_repo": "HlexNC/chexvision-demo"
  },
  "model": {
    "type": "scratch",
    "name": "CheXVision-ResNet",
    "architecture": {
      "input_channels": 3,
      "initial_filters": 64,
      "block_config": [
        3,
        4,
        6,
        3
      ],
      "filter_sizes": [
        64,
        128,
        256,
        512
      ],
      "dropout": 0.5,
      "activation": "relu",
      "use_batch_norm": true,
      "global_avg_pool": true,
      "use_se": true
    },
    "heads": {
      "multilabel": {
        "num_classes": 14,
        "activation": "sigmoid"
      },
      "binary": {
        "num_classes": 1,
        "activation": "sigmoid"
      }
    }
  }
}