config.json

{
  "model_type": "resnet",
  "num_labels": 2,
  "labels": {
    "0": "diagram",
    "1": "none"
  },
  "input_size": [224, 224],
  "normalization": {
    "mean": [0.485, 0.456, 0.406],
    "std": [0.229, 0.224, 0.225]
  },
  "framework": "pytorch"
}