model_config.json

{
  "arch": "efficientnet",
  "num_classes": 10,
  "img_size": 128
}