| { |
| "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": 60, |
| "batch_size": 24, |
| "learning_rate": 0.0001, |
| "weight_decay": 0.0001, |
| "optimizer": "adamw", |
| "scheduler": "cosine", |
| "warmup_epochs": 3, |
| "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": "densenet", |
| "name": "CheXVision-DenseNet", |
| "architecture": { |
| "backbone": "densenet121", |
| "pretrained": true, |
| "freeze_features": false, |
| "dropout": 0.3 |
| }, |
| "fine_tuning": { |
| "freeze_epochs": 5, |
| "freeze_lr": 0.001, |
| "unfreeze_lr": 0.0001 |
| }, |
| "heads": { |
| "multilabel": { |
| "num_classes": 14, |
| "activation": "sigmoid" |
| }, |
| "binary": { |
| "num_classes": 1, |
| "activation": "sigmoid" |
| } |
| } |
| } |
| } |
