File size: 2,475 Bytes
7155499 819bc96 7155499 08c8aa7 7155499 08c8aa7 7155499 819bc96 7155499 08c8aa7 819bc96 7155499 08c8aa7 7155499 08c8aa7 7155499 819bc96 7155499 08c8aa7 7155499 08c8aa7 7155499 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 | {
"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"
}
}
}
} |