config.json

{
  "model_type": "rexnet",
  "architecture": "rexnet_150",
  "num_classes": 7,
  "input_size": [224, 224],
  "class_names": ["Invalid", "SDTI", "Stage_I", "Stage_II", "Stage_III", "Stage_IV", "Unstageable"],
  "task": "image-classification",
  "framework": "pytorch",
  "custom_handler": true
}