| { | |
| "framework": "jax", | |
| "model_type": "canvit_classification", | |
| "model_config": { | |
| "embed_dim": 768, | |
| "num_heads": 12, | |
| "n_blocks": 12, | |
| "patch_size": 16, | |
| "ffn_ratio": 4.0, | |
| "rope_base": 100.0, | |
| "layerscale_init": 1e-05, | |
| "rw_stride": 2, | |
| "enable_reads": true, | |
| "n_backbone_registers": 5, | |
| "n_canvas_registers": 16, | |
| "canvas_num_heads": 8, | |
| "canvas_head_dim": 128, | |
| "enable_vpe": true | |
| }, | |
| "n_classes": 1000, | |
| "source": "canvit/canvitb16-add-vpe-finetune-g128px-s512px-in1k-2026-04-06" | |
| } |