ai-image-det-resnet152 / config.json
RishiDarkDevil's picture
Upload feature extractor (#2)
e44be29
{
"architectures": [
"ResnetFeatureExtractor"
],
"auto_map": {
"AutoConfig": "configuration_resnet.ResnetFeatureExtractorConfig",
"AutoModel": "model_resnet.ResnetFeatureExtractor",
"AutoModelForImageClassification": "model_resnet.ResnetModelForImageClassification"
},
"model_type": "resnet",
"name": "resnet152",
"torch_dtype": "float32",
"transformers_version": "4.27.1"
}