rwightman's picture
rwightman HF staff
Update model config and README
bab93e2
{
"architecture": "maxvit_base_tf_512",
"num_classes": 1000,
"num_features": 768,
"global_pool": "avg",
"pretrained_cfg": {
"tag": "in1k",
"custom_load": false,
"input_size": [
3,
512,
512
],
"fixed_input_size": true,
"interpolation": "bicubic",
"crop_pct": 1.0,
"crop_mode": "squash",
"mean": [
0.5,
0.5,
0.5
],
"std": [
0.5,
0.5,
0.5
],
"num_classes": 1000,
"pool_size": [
16,
16
],
"first_conv": "stem.conv1",
"classifier": "head.fc"
}
}