config.json

{
  "architectures": [
    "MyModel"
  ],
  "auto_map": {
    "AutoConfig": "configuration_my.MyConfig",
    "AutoModel": "modeling_my.MyModel"
  },
  "avg_down": false,
  "base_width": 64,
  "block_type": "bottleneck",
  "cardinality": 1,
  "input_channels": 3,
  "layers": [
    3,
    4,
    6,
    3
  ],
  "model_type": "my",
  "num_classes": 1000,
  "stem_type": "",
  "stem_width": 64,
  "torch_dtype": "float32",
  "transformers_version": "4.28.1"
}