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