config.json

{
  "architecture": "vgg_model",
  "num_classes": 10,
  "num_features": 4096,
  "pretrained_cfg": {
    "crop_mode": "center",
    "first_conv": "features[0]",
    "classifier": "classifier[6]",
    "input_size": [
      3,
      32,
      32
    ],
    "num_classes": 10,
    "num_features": 4096,
    "mean": [
      0.5071,
      0.4865,
      0.4409
    ],
    "std": [
      0.2673,
      0.2564,
      0.2762
    ]
  }
}