{ "architecture": "tiny_vit_21m_512", "num_classes": 1000, "num_features": 576, "pretrained_cfg": { "tag": "dist_in22k_ft_in1k", "custom_load": false, "input_size": [ 3, 512, 512 ], "fixed_input_size": false, "interpolation": "bicubic", "crop_pct": 1.0, "crop_mode": "squash", "mean": [ 0.485, 0.456, 0.406 ], "std": [ 0.229, 0.224, 0.225 ], "num_classes": 1000, "pool_size": [ 16, 16 ], "first_conv": "patch_embed.conv1.conv", "classifier": "head.fc" } }