custom-tvres18 / config.json
wucng's picture
Upload config
7737c33 verified
{
"_name_or_path": "custom-tvres18",
"architectures": [
"ResnetModel"
],
"input_channels": 3,
"model_name": "resnet18",
"model_type": "tvresnet",
"num_classes": 1000,
"pretrained": true,
"torch_dtype": "float32",
"transformers_version": "4.44.0"
}