config.json

{
  "add_head": true,
  "dims": [
    768,
    512
  ],
  "name": "vit_base_patch32_384.augreg_in1k",
  "pool": "gem",
  "pool_param": 3.0,
  "pretrained": null,
  "use_classify": false
}