File size: 986 Bytes
649f919 |
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 |
{
"model_type": "googlenet",
"model_name": "Googlenet_pneumonia_binaryclassification",
"architecture": "googlenet",
"task": "image-classification",
"num_labels": 2,
"label2id": {
"Normal": 0,
"Pneumonia": 1
},
"id2label": {
"0": "Normal",
"1": "Pneumonia"
},
"input_size": [224, 224],
"pretrained": true,
"weights": "models.GoogLeNet_Weights.DEFAULT",
"classifier_dropout": 0.5,
"batch_size": 256,
"epochs": 20,
"optimizer": "Adam",
"scheduler": "ReduceLROnPlateau",
"learning_rate": {
"features_lr": 1e-4,
"classifier_lr": 1e-4
},
"loss_function": "CrossEntropyLoss",
"label_smoothing": 0.1,
"augmentation": {
"resize": [224, 224],
"random_flip": 0.3,
"random_affine": {
"degrees": [-5, 5],
"translate": [0.05, 0.05],
"scale": [0.95, 1.05]
},
"random_color_jitter": {
"brightness": 0.2,
"contrast": 0.2
}
},
"early_stopping_patience": 3,
"device": "cuda"
}
|