config.json

{
    "model_type": "efficientnet",
    "model_name": "efficientnet-b0",
    "pretrained": true,
    "num_classes": 101,
    "image_size": 224,
    "input_channels": 3,
    "optimizer": "Adam",
    "learning_rate": 0.001,
    "epoch": 10,
    "loss_function": "CrossEntropyLoss",
    "normalize_mean": [0.485, 0.456, 0.406],
    "normalize_std": [0.229, 0.224, 0.225]
}