| { | |
| "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" | |
| } | |
| } | |
| } | |
| } |