custom-resnet50d / config.json
NexaAI's picture
Upload model
d529765 verified
{
"architectures": [
"ResnetModelForImageClassification"
],
"avg_down": true,
"base_width": 64,
"block_type": "bottleneck",
"cardinality": 1,
"input_channels": 3,
"layers": [
3,
4,
6,
3
],
"model_type": "resnet",
"num_classes": 1000,
"stem_type": "deep",
"stem_width": 32,
"torch_dtype": "float32",
"transformers_version": "4.36.2"
}