content/config.json

{
  "architecture": "resnet18",
  "framework": "pytorch",
  "task": "image-classification",
  "num_classes": 10,
  "classifier": {
    "type": "Sequential",
    "layers": [
      {
        "Linear": [
          512,
          120
        ]
      },
      "ReLU",
      {
        "Dropout": 0.2
      },
      {
        "Linear": [
          120,
          10
        ]
      }
    ]
  },
  "input_size": [
    3,
    224,
    224
  ]
}