custom-resnet50d / config.json
david-hcl's picture
Updating model weights
31f9a95 verified
{
"architectures": [
"ResnetModelForImageClassification"
],
"auto_map": {
"AutoConfig": "configuration_resnet.ResnetConfig",
"AutoModelForImageClassification": "modeling_resnet.ResnetModelForImageClassification"
},
"avg_down": true,
"base_width": 64,
"block_type": "bottleneck",
"cardinality": 1,
"depth": 50,
"layers": [
3,
4,
6,
3
],
"model_type": "resnet",
"new_attribute": "force_update",
"num_channels": 3,
"num_classes": 1000,
"stem_type": "deep",
"stem_width": 32,
"torch_dtype": "float32",
"transformers_version": "4.49.0.dev0"
}