gaunernst's picture
Create config.json
1d0bc2d verified
{
"architecture": "convnext_atto",
"num_classes": 512,
"num_features": 320,
"pretrained_cfg": {
"input_size": [3, 112, 112],
"fixed_input_size": false,
"interpolation": "bicubic",
"crop_pct": 1.0,
"crop_mode": "center",
"mean": [0.5, 0.5, 0.5],
"std": [0.5, 0.5, 0.5],
"num_classes": 512,
"pool_size": [7, 7],
"first_conv": "stem.0",
"classifier": "head.fc"
},
"model_args": {
"patch_size": 2
}
}