1aurent's picture
Update config.json
41f882e
{
"architecture": "vit_base_patch16_224",
"num_classes": 0,
"num_features": 768,
"global_pool": "token",
"pretrained_cfg": {
"tag": "owkin_pancancer",
"custom_load": false,
"input_size": [
3,
224,
224
],
"fixed_input_size": true,
"interpolation": "bicubic",
"crop_pct": 1,
"crop_mode": "center",
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
],
"pool_size": null,
"first_conv": "patch_embed.proj",
"classifier": "head"
}
}