config.json

{
    "model_type": "custom-resnet18", 
    "num_classes": 7,
    "hidden_size": 512,
    "image_size": [100, 100],
    "normalize_mean": [0.485, 0.456, 0.406],
    "normalize_std": [0.229, 0.224, 0.225],
    "activation": "PReLU",
    "base_model": "resnet18"

  }