custom-my / config.json
MLFarmer's picture
Upload model
79ba3b2
{
"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"
}