config.json

{
    "model_type": "resnet18",
    "pretrained": false,
    "num_classes": 10, 
    "fc_in_features": 512  
  }