config.json

{
  "avg_pool": 1,
  "num_classes": 5
}