| { | |
| "model_type": "densenet", | |
| "model_name": "Densenet169_pnuemonia_binaryclassification", | |
| "architecture": "densenet169", | |
| "task": "image-classification", | |
| "num_labels": 2, | |
| "label2id": { | |
| "Normal": 0, | |
| "Pneumonia": 1 | |
| }, | |
| "id2label": { | |
| "0": "Normal", | |
| "1": "Pneumonia" | |
| }, | |
| "input_size": [224, 224], | |
| "pretrained": true, | |
| "weights": "models.DenseNet169_Weights.DEFAULT", | |
| "classifier_dropout": 0.4, | |
| "batch_size": 128, | |
| "epochs": 30, | |
| "optimizer": "Lion", | |
| "scheduler": "OneCycleLR", | |
| "learning_rate": { | |
| "features_lr": 7e-5, | |
| "classifier_lr": 1e-4 | |
| }, | |
| "loss_function": "FocalLoss", | |
| "focal_loss_params": { | |
| "alpha": 2.5, | |
| "gamma": 2 | |
| }, | |
| "augmentation": { | |
| "resize": [224, 224], | |
| "random_flip": 0.3, | |
| "random_affine": { | |
| "degrees": [-15, 15], | |
| "translate": [0.1, 0.1], | |
| "scale": [0.85, 1.15] | |
| }, | |
| "random_color_jitter": { | |
| "brightness": 0.2, | |
| "contrast": 0.2 | |
| }, | |
| "random_blur": 0.2, | |
| "random_erasing": 0.1 | |
| }, | |
| "early_stopping_patience": 7, | |
| "device": "cuda" | |
| } | |