Full list of available pretrained weights

I listed all the weights that are now available in ~/.cache/torch/hub/checkpoints:

alexnet-owt-7be5be79.pth
convnext_base-6075fbad.pth
convnext_large-ea097f82.pth
convnext_small-0c510722.pth
convnext_tiny-983f1562.pth
deeplabv3_mobilenet_v3_large-fc3c493d.pth
deeplabv3_resnet101_coco-586e9e4e.pth
deeplabv3_resnet50_coco-cd0a2569.pth
densenet121-a639ec97.pth
densenet161-8d451a50.pth
densenet169-b2777c0a.pth
densenet201-c1103571.pth
efficientnet_b0_rwightman-3dd342df.pth
efficientnet_b1-c27df63c.pth
efficientnet_b1_rwightman-533bc792.pth
efficientnet_b2_rwightman-bcdf34b7.pth
efficientnet_b3_rwightman-cf984f9c.pth
efficientnet_b4_rwightman-7eb33cd5.pth
efficientnet_b5_lukemelas-b6417697.pth
efficientnet_b6_lukemelas-c76e70fd.pth
efficientnet_b7_lukemelas-dcc49843.pth
fasterrcnn_mobilenet_v3_large_320_fpn-907ea3f9.pth
fasterrcnn_mobilenet_v3_large_fpn-fb6a3cc7.pth
fasterrcnn_resnet50_fpn_coco-258fb6c6.pth
fcn_resnet101_coco-7ecb50ca.pth
fcn_resnet50_coco-1167a1af.pth
fcos_resnet50_fpn_coco-99b0c9b7.pth
googlenet-1378be20.pth
googlenet_fbgemm-c00238cf.pth
inception_v3_google-0cc3c7bd.pth
inception_v3_google_fbgemm-71447a44.pth
keypointrcnn_resnet50_fpn_coco-9f466800.pth
keypointrcnn_resnet50_fpn_coco-fc266e95.pth
lraspp_mobilenet_v3_large-d234d4ea.pth
maskrcnn_resnet50_fpn_coco-bf2d0c1e.pth
mc3_18-a90a0ba3.pth
mnasnet0.5_top1_67.823-3ffadce67e.pth
mnasnet1.0_top1_73.512-f206786ef8.pth
mobilenet_v2-b0353104.pth
mobilenet_v2_qnnpack_37f702c5.pth
mobilenet_v3_large-5c1a4163.pth
mobilenet_v3_large-8738ca79.pth
mobilenet_v3_large_qnnpack-5bcacf28.pth
mobilenet_v3_small-047dcff4.pth
r2plus1d_18-91a641e6.pth
r3d_18-b3b3357e.pth
raft_large_C_T_SKHT_K_V1-4a6a5039.pth
raft_large_C_T_SKHT_K_V2-b5c70766.pth
raft_large_C_T_SKHT_V1-0b8c9e55.pth
raft_large_C_T_SKHT_V2-ff5fadd5.pth
raft_large_C_T_V1-22a6c225.pth
raft_large_C_T_V2-1bb1363a.pth
raft_small_C_T_V1-ad48884c.pth
raft_small_C_T_V2-01064c6d.pth
regnet_x_16gf-2007eb11.pth
regnet_x_16gf-ba3796d7.pth
regnet_x_1_6gf-a12f2b72.pth
regnet_x_1_6gf-e3633e7f.pth
regnet_x_32gf-6eb8fdc6.pth
regnet_x_32gf-9d47f8d0.pth
regnet_x_3_2gf-7071aa85.pth
regnet_x_3_2gf-f342aeae.pth
regnet_x_400mf-62229a5f.pth
regnet_x_400mf-adf1edd5.pth
regnet_x_800mf-94a99ebd.pth
regnet_x_800mf-ad17e45c.pth
regnet_x_8gf-03ceed89.pth
regnet_x_8gf-2b70d774.pth
regnet_y_16gf-3e4a00f9.pth
regnet_y_16gf-9e6ed7dd.pth
regnet_y_1_6gf-0d7bc02a.pth
regnet_y_1_6gf-b11a554e.pth
regnet_y_32gf-4dee3f7a.pth
regnet_y_32gf-8db6d4b5.pth
regnet_y_3_2gf-9180c971.pth
regnet_y_3_2gf-b5a9779c.pth
regnet_y_400mf-c65dace8.pth
regnet_y_400mf-e6988f5f.pth
regnet_y_800mf-1b27b58c.pth
regnet_y_800mf-58fc7688.pth
regnet_y_8gf-d0d0e4a8.pth
regnet_y_8gf-dc2b1b54.pth
resnet101-63fe2227.pth
resnet101-cd907fc2.pth
resnet152-394f9c45.pth
resnet152-f82ba261.pth
resnet18-f37072fd.pth
resnet18_fbgemm_16fa66dd.pth
resnet34-b627a593.pth
resnet50-0676ba61.pth
resnet50-11ad3fa6.pth
resnet50_fbgemm-23753f79.pth
resnet50_fbgemm_bf931d71.pth
resnext101_32x8_fbgemm-ee16d00c.pth
resnext101_32x8_fbgemm_09835ccf.pth
resnext101_32x8d-110c445d.pth
resnext101_32x8d-8ba56ff5.pth
resnext50_32x4d-1a0047aa.pth
resnext50_32x4d-7cdf4587.pth
retinanet_resnet50_fpn_coco-eeacb38b.pth
shufflenetv2_x0.5-f707e7126e.pth
shufflenetv2_x0.5_fbgemm-00845098.pth
shufflenetv2_x1-5666bf0f80.pth
shufflenetv2_x1_fbgemm-db332c57.pth
squeezenet1_0-b66bff10.pth
squeezenet1_1-b8a52dc0.pth
ssd300_vgg16_coco-b556d3b4.pth
ssdlite320_mobilenet_v3_large_coco-a79551df.pth
vgg11-8a719046.pth
vgg11_bn-6002323d.pth
vgg13-19584684.pth
vgg13_bn-abd245e5.pth
vgg16-397923af.pth
vgg16_bn-6c64b313.pth
vgg16_features-amdegroot-88682ab5.pth
vgg16_reducedfc.pth
vgg19-dcbb9e9d.pth
vgg19_bn-c79401a0.pth
vit_b_16-c867db91.pth
vit_b_32-d86f8d99.pth
vit_l_16-852ce7e3.pth
vit_l_32-c7638314.pth
wide_resnet101_2-32ee1156.pth
wide_resnet101_2-d733dc28.pth
wide_resnet50_2-95faca4d.pth
wide_resnet50_2-9ba9bcbe.pth
7 Likes

wow, which pytorch version that got vit? nightly version?

yes, nightly version has it. However the accuracies they report for these vit weights are not especially great:
https://pytorch.org/vision/main/models.html