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