config.json

{
    "model_name": "ResNet18",
    "input_size": [3, 128, 128],
    "output_size": 2,
    "class_names": ["Cat", "Dog"],
    "mean": [0.485, 0.456, 0.406],
    "std": [0.229, 0.224, 0.225]
}