resnet50 / config.json
leftthomas's picture
Update config.json
c8189a8
{
"architectures": [
"ResnetModelForImageClassification"
],
"auto_map": {
"AutoConfig": "configuration_resnet.ResnetConfig",
"AutoModelForImageClassification": "modeling_resnet.ResnetModelForImageClassification"
},
"block_type": "bottleneck",
"layers": [
3,
4,
6,
3
],
"model_type": "resnet",
"num_classes": 1000,
"torch_dtype": "float32",
"transformers_version": "4.17.0"
}