config.json

{
  "architectures": [
    "ResNetForImageClassification"
  ],
  "depths": [3, 4, 6, 3],
  "downsample_in_first_stage": false,
  "embedding_size": 64,
  "hidden_act": "relu",
  "hidden_sizes": [256, 512, 1024, 2048],

  "layer_type": "bottleneck",
  "model_type": "resnet",
  "num_channels": 3
}