config.json

{
  "architectures": ["TinyCNN"],
  "num_channels": 3,
  "image_size": 224,
  "num_labels": 10,
  "model_type": "torchscript",
  "library_name": "pytorch"
}