rwightman Models

rwightman maintains an awesome (700+ and growing!) collection of models that is published via torch.hub - see the list below! Typically, you’ll want to load these pre-trained, in order to use them with your own dataset (and your own number of classes). In that case you should use Pywick’s models.model_utils.get_model(...) utility function which will do all the dirty work for you and give you a pretrained model but with your custom number of classes!

  • DPN
  • dpn68
  • dpn68b
  • dpn98
  • dpn131
  • dpn107
  • BNInception
  • bninception
  • FBResNet
  • FBResNet18
  • FBResNet34
  • FBResNet50
  • FBResNet101
  • fbresnet152
  • InceptionResV2
  • InceptionResNetV2
  • inceptionresnetv2
  • InceptionV4
  • inceptionv4
  • nasnetalarge
  • NASNetALarge
  • nasnetamobile
  • NASNetAMobile
  • pnasnet5large
  • PNASNet5Large
  • PolyNet
  • polynet
  • PyResNet18
  • PyResNet34
  • PyResNet
  • PreactResnet110
  • PreactResnet164_bottleneck
  • ResNet_swish
  • ResNet18_swish
  • ResNet34_swish
  • ResNet50_swish
  • ResNet101_swish
  • ResNet152_swish
  • ResNeXt50_32x4d
  • resnext50_32x4d
  • ResNeXt101_32x4d
  • resnext101_32x4d
  • ResNeXt101_64x4d
  • resnext101_64x4d
  • SENet
  • senet154
  • se_resnet50
  • se_resnet101
  • se_resnet152
  • se_resnext50_32x4d
  • se_resnext101_32x4d
  • WideResNet
  • wideresnet50
  • Xception
  • xception
  • se_densenet121
  • se_densenet161
  • se_densenet169
  • se_densenet201
  • AlexNet
  • ResNet
  • resnet18
  • resnet34
  • resnet50
  • resnet101
  • resnet152
  • resnext50_32x4d
  • resnext101_32x8d
  • wide_resnet50_2
  • wide_resnet101_2
  • VGG
  • vgg11
  • vgg11_bn
  • vgg13
  • vgg13_bn
  • vgg16
  • vgg16_bn
  • vgg19_bn
  • vgg19
  • SqueezeNet
  • squeezenet1_0
  • squeezenet1_1
  • Inception3
  • inception_v3
  • InceptionOutputs
  • _InceptionOutputs
  • DenseNet
  • densenet121
  • densenet169
  • densenet201
  • densenet161
  • GoogLeNet
  • GoogLeNetOutputs
  • _GoogLeNetOutputs
  • MobileNetV2
  • mobilenet_v2
  • MNASNet
  • mnasnet0_5
  • mnasnet0_75
  • mnasnet1_0
  • mnasnet1_3
  • ShuffleNetV2
  • shufflenet_v2_x0_5
  • shufflenet_v2_x1_0
  • shufflenet_v2_x1_5
  • shufflenet_v2_x2_0
  • adv_inception_v3
  • bat_resnext26ts
  • beit_base_patch16_224
  • beit_base_patch16_224_in22k
  • beit_base_patch16_384
  • beit_large_patch16_224
  • beit_large_patch16_224_in22k
  • beit_large_patch16_384
  • beit_large_patch16_512
  • botnet26t_256
  • botnet50ts_256
  • cait_m36_384
  • cait_m48_448
  • cait_s24_224
  • cait_s24_384
  • cait_s36_384
  • cait_xs24_384
  • cait_xxs24_224
  • cait_xxs24_384
  • cait_xxs36_224
  • cait_xxs36_384
  • coat_lite_mini
  • coat_lite_small
  • coat_lite_tiny
  • coat_mini
  • coat_tiny
  • convit_base
  • convit_small
  • convit_tiny
  • crossvit_15_240
  • crossvit_15_dagger_240
  • crossvit_15_dagger_408
  • crossvit_18_240
  • crossvit_18_dagger_240
  • crossvit_18_dagger_408
  • crossvit_9_240
  • crossvit_9_dagger_240
  • crossvit_base_240
  • crossvit_small_240
  • crossvit_tiny_240
  • cspdarknet53
  • cspdarknet53_iabn
  • cspresnet50
  • cspresnet50d
  • cspresnet50w
  • cspresnext50
  • cspresnext50_iabn
  • darknet53
  • deit_base_distilled_patch16_224
  • deit_base_distilled_patch16_384
  • deit_base_patch16_224
  • deit_base_patch16_384
  • deit_small_distilled_patch16_224
  • deit_small_patch16_224
  • deit_tiny_distilled_patch16_224
  • deit_tiny_patch16_224
  • densenet121
  • densenet121d
  • densenet161
  • densenet169
  • densenet201
  • densenet264
  • densenet264d_iabn
  • densenetblur121d
  • dla102
  • dla102x
  • dla102x2
  • dla169
  • dla34
  • dla46_c
  • dla46x_c
  • dla60
  • dla60_res2net
  • dla60_res2next
  • dla60x
  • dla60x_c
  • dm_nfnet_f0
  • dm_nfnet_f1
  • dm_nfnet_f2
  • dm_nfnet_f3
  • dm_nfnet_f4
  • dm_nfnet_f5
  • dm_nfnet_f6
  • dpn107
  • dpn131
  • dpn68
  • dpn68b
  • dpn92
  • dpn98
  • eca_botnext26ts_256
  • eca_halonext26ts
  • eca_nfnet_l0
  • eca_nfnet_l1
  • eca_nfnet_l2
  • eca_nfnet_l3
  • eca_resnet33ts
  • eca_resnext26ts
  • eca_vovnet39b
  • ecaresnet101d
  • ecaresnet101d_pruned
  • ecaresnet200d
  • ecaresnet269d
  • ecaresnet26t
  • ecaresnet50d
  • ecaresnet50d_pruned
  • ecaresnet50t
  • ecaresnetlight
  • ecaresnext26t_32x4d
  • ecaresnext50t_32x4d
  • efficientnet_b0
  • efficientnet_b1
  • efficientnet_b1_pruned
  • efficientnet_b2
  • efficientnet_b2_pruned
  • efficientnet_b2a
  • efficientnet_b3
  • efficientnet_b3_pruned
  • efficientnet_b3a
  • efficientnet_b4
  • efficientnet_b5
  • efficientnet_b6
  • efficientnet_b7
  • efficientnet_b8
  • efficientnet_cc_b0_4e
  • efficientnet_cc_b0_8e
  • efficientnet_cc_b1_8e
  • efficientnet_el
  • efficientnet_el_pruned
  • efficientnet_em
  • efficientnet_es
  • efficientnet_es_pruned
  • efficientnet_l2
  • efficientnet_lite0
  • efficientnet_lite1
  • efficientnet_lite2
  • efficientnet_lite3
  • efficientnet_lite4
  • efficientnetv2_l
  • efficientnetv2_m
  • efficientnetv2_rw_m
  • efficientnetv2_rw_s
  • efficientnetv2_rw_t
  • efficientnetv2_s
  • efficientnetv2_xl
  • ens_adv_inception_resnet_v2
  • ese_vovnet19b_dw
  • ese_vovnet19b_slim
  • ese_vovnet19b_slim_dw
  • ese_vovnet39b
  • ese_vovnet39b_evos
  • ese_vovnet57b
  • ese_vovnet99b
  • ese_vovnet99b_iabn
  • fbnetc_100
  • fbnetv3_b
  • fbnetv3_d
  • fbnetv3_g
  • gc_efficientnetv2_rw_t
  • gcresnet33ts
  • gcresnet50t
  • gcresnext26ts
  • gcresnext50ts
  • gernet_l
  • gernet_m
  • gernet_s
  • ghostnet_050
  • ghostnet_100
  • ghostnet_130
  • gluon_inception_v3
  • gluon_resnet101_v1b
  • gluon_resnet101_v1c
  • gluon_resnet101_v1d
  • gluon_resnet101_v1s
  • gluon_resnet152_v1b
  • gluon_resnet152_v1c
  • gluon_resnet152_v1d
  • gluon_resnet152_v1s
  • gluon_resnet18_v1b
  • gluon_resnet34_v1b
  • gluon_resnet50_v1b
  • gluon_resnet50_v1c
  • gluon_resnet50_v1d
  • gluon_resnet50_v1s
  • gluon_resnext101_32x4d
  • gluon_resnext101_64x4d
  • gluon_resnext50_32x4d
  • gluon_senet154
  • gluon_seresnext101_32x4d
  • gluon_seresnext101_64x4d
  • gluon_seresnext50_32x4d
  • gluon_xception65
  • gmixer_12_224
  • gmixer_24_224
  • gmlp_b16_224
  • gmlp_s16_224
  • gmlp_ti16_224
  • halonet26t
  • halonet50ts
  • halonet_h1
  • hardcorenas_a
  • hardcorenas_b
  • hardcorenas_c
  • hardcorenas_d
  • hardcorenas_e
  • hardcorenas_f
  • hrnet_w18
  • hrnet_w18_small
  • hrnet_w18_small_v2
  • hrnet_w30
  • hrnet_w32
  • hrnet_w40
  • hrnet_w44
  • hrnet_w48
  • hrnet_w64
  • ig_resnext101_32x16d
  • ig_resnext101_32x32d
  • ig_resnext101_32x48d
  • ig_resnext101_32x8d
  • inception_resnet_v2
  • inception_v3
  • inception_v4
  • jx_nest_base
  • jx_nest_small
  • jx_nest_tiny
  • lambda_resnet26t
  • legacy_senet154
  • legacy_seresnet101
  • legacy_seresnet152
  • legacy_seresnet18
  • legacy_seresnet34
  • legacy_seresnet50
  • legacy_seresnext101_32x4d
  • legacy_seresnext26_32x4d
  • legacy_seresnext50_32x4d
  • levit_128
  • levit_128s
  • levit_192
  • levit_256
  • levit_384
  • mixer_b16_224
  • mixer_b16_224_in21k
  • mixer_b16_224_miil
  • mixer_b16_224_miil_in21k
  • mixer_b32_224
  • mixer_l16_224
  • mixer_l16_224_in21k
  • mixer_l32_224
  • mixer_s16_224
  • mixer_s32_224
  • mixnet_l
  • mixnet_m
  • mixnet_s
  • mixnet_xl
  • mixnet_xxl
  • mnasnet_050
  • mnasnet_075
  • mnasnet_100
  • mnasnet_140
  • mnasnet_a1
  • mnasnet_b1
  • mnasnet_small
  • mobilenetv2_100
  • mobilenetv2_110d
  • mobilenetv2_120d
  • mobilenetv2_140
  • mobilenetv3_large_075
  • mobilenetv3_large_100
  • mobilenetv3_large_100_miil
  • mobilenetv3_large_100_miil_in21k
  • mobilenetv3_rw
  • mobilenetv3_small_075
  • mobilenetv3_small_100
  • nasnetalarge
  • nest_base
  • nest_small
  • nest_tiny
  • nf_ecaresnet101
  • nf_ecaresnet26
  • nf_ecaresnet50
  • nf_regnet_b0
  • nf_regnet_b1
  • nf_regnet_b2
  • nf_regnet_b3
  • nf_regnet_b4
  • nf_regnet_b5
  • nf_resnet101
  • nf_resnet26
  • nf_resnet50
  • nf_seresnet101
  • nf_seresnet26
  • nf_seresnet50
  • nfnet_f0
  • nfnet_f0s
  • nfnet_f1
  • nfnet_f1s
  • nfnet_f2
  • nfnet_f2s
  • nfnet_f3
  • nfnet_f3s
  • nfnet_f4
  • nfnet_f4s
  • nfnet_f5
  • nfnet_f5s
  • nfnet_f6
  • nfnet_f6s
  • nfnet_f7
  • nfnet_f7s
  • nfnet_l0
  • pit_b_224
  • pit_b_distilled_224
  • pit_s_224
  • pit_s_distilled_224
  • pit_ti_224
  • pit_ti_distilled_224
  • pit_xs_224
  • pit_xs_distilled_224
  • pnasnet5large
  • regnetx_002
  • regnetx_004
  • regnetx_006
  • regnetx_008
  • regnetx_016
  • regnetx_032
  • regnetx_040
  • regnetx_064
  • regnetx_080
  • regnetx_120
  • regnetx_160
  • regnetx_320
  • regnety_002
  • regnety_004
  • regnety_006
  • regnety_008
  • regnety_016
  • regnety_032
  • regnety_040
  • regnety_064
  • regnety_080
  • regnety_120
  • regnety_160
  • regnety_320
  • repvgg_a2
  • repvgg_b0
  • repvgg_b1
  • repvgg_b1g4
  • repvgg_b2
  • repvgg_b2g4
  • repvgg_b3
  • repvgg_b3g4
  • res2net101_26w_4s
  • res2net50_14w_8s
  • res2net50_26w_4s
  • res2net50_26w_6s
  • res2net50_26w_8s
  • res2net50_48w_2s
  • res2next50
  • resmlp_12_224
  • resmlp_12_distilled_224
  • resmlp_24_224
  • resmlp_24_distilled_224
  • resmlp_36_224
  • resmlp_36_distilled_224
  • resmlp_big_24_224
  • resmlp_big_24_224_in22ft1k
  • resmlp_big_24_distilled_224
  • resnest101e
  • resnest14d
  • resnest200e
  • resnest269e
  • resnest26d
  • resnest50d
  • resnest50d_1s4x24d
  • resnest50d_4s2x40d
  • resnet101
  • resnet101d
  • resnet152
  • resnet152d
  • resnet18
  • resnet18d
  • resnet200
  • resnet200d
  • resnet26
  • resnet26d
  • resnet26t
  • resnet32ts
  • resnet33ts
  • resnet34
  • resnet34d
  • resnet50
  • resnet50d
  • resnet50t
  • resnet51q
  • resnet61q
  • resnetblur18
  • resnetblur50
  • resnetrs101
  • resnetrs152
  • resnetrs200
  • resnetrs270
  • resnetrs350
  • resnetrs420
  • resnetrs50
  • resnetv2_101
  • resnetv2_101d
  • resnetv2_101x1_bitm
  • resnetv2_101x1_bitm_in21k
  • resnetv2_101x3_bitm
  • resnetv2_101x3_bitm_in21k
  • resnetv2_152
  • resnetv2_152d
  • resnetv2_152x2_bit_teacher
  • resnetv2_152x2_bit_teacher_384
  • resnetv2_152x2_bitm
  • resnetv2_152x2_bitm_in21k
  • resnetv2_152x4_bitm
  • resnetv2_152x4_bitm_in21k
  • resnetv2_50
  • resnetv2_50d
  • resnetv2_50t
  • resnetv2_50x1_bit_distilled
  • resnetv2_50x1_bitm
  • resnetv2_50x1_bitm_in21k
  • resnetv2_50x3_bitm
  • resnetv2_50x3_bitm_in21k
  • resnext101_32x4d
  • resnext101_32x8d
  • resnext101_64x4d
  • resnext26ts
  • resnext50_32x4d
  • resnext50d_32x4d
  • rexnet_100
  • rexnet_130
  • rexnet_150
  • rexnet_200
  • rexnetr_100
  • rexnetr_130
  • rexnetr_150
  • rexnetr_200
  • sehalonet33ts
  • selecsls42
  • selecsls42b
  • selecsls60
  • selecsls60b
  • selecsls84
  • semnasnet_050
  • semnasnet_075
  • semnasnet_100
  • semnasnet_140
  • senet154
  • seresnet101
  • seresnet152
  • seresnet152d
  • seresnet18
  • seresnet200d
  • seresnet269d
  • seresnet33ts
  • seresnet34
  • seresnet50
  • seresnet50t
  • seresnext101_32x4d
  • seresnext101_32x8d
  • seresnext26d_32x4d
  • seresnext26t_32x4d
  • seresnext26tn_32x4d
  • seresnext26ts
  • seresnext50_32x4d
  • skresnet18
  • skresnet34
  • skresnet50
  • skresnet50d
  • skresnext50_32x4d
  • spnasnet_100
  • ssl_resnet18
  • ssl_resnet50
  • ssl_resnext101_32x16d
  • ssl_resnext101_32x4d
  • ssl_resnext101_32x8d
  • ssl_resnext50_32x4d
  • swin_base_patch4_window12_384
  • swin_base_patch4_window12_384_in22k
  • swin_base_patch4_window7_224
  • swin_base_patch4_window7_224_in22k
  • swin_large_patch4_window12_384
  • swin_large_patch4_window12_384_in22k
  • swin_large_patch4_window7_224
  • swin_large_patch4_window7_224_in22k
  • swin_small_patch4_window7_224
  • swin_tiny_patch4_window7_224
  • swsl_resnet18
  • swsl_resnet50
  • swsl_resnext101_32x16d
  • swsl_resnext101_32x4d
  • swsl_resnext101_32x8d
  • swsl_resnext50_32x4d
  • tf_efficientnet_b0
  • tf_efficientnet_b0_ap
  • tf_efficientnet_b0_ns
  • tf_efficientnet_b1
  • tf_efficientnet_b1_ap
  • tf_efficientnet_b1_ns
  • tf_efficientnet_b2
  • tf_efficientnet_b2_ap
  • tf_efficientnet_b2_ns
  • tf_efficientnet_b3
  • tf_efficientnet_b3_ap
  • tf_efficientnet_b3_ns
  • tf_efficientnet_b4
  • tf_efficientnet_b4_ap
  • tf_efficientnet_b4_ns
  • tf_efficientnet_b5
  • tf_efficientnet_b5_ap
  • tf_efficientnet_b5_ns
  • tf_efficientnet_b6
  • tf_efficientnet_b6_ap
  • tf_efficientnet_b6_ns
  • tf_efficientnet_b7
  • tf_efficientnet_b7_ap
  • tf_efficientnet_b7_ns
  • tf_efficientnet_b8
  • tf_efficientnet_b8_ap
  • tf_efficientnet_cc_b0_4e
  • tf_efficientnet_cc_b0_8e
  • tf_efficientnet_cc_b1_8e
  • tf_efficientnet_el
  • tf_efficientnet_em
  • tf_efficientnet_es
  • tf_efficientnet_l2_ns
  • tf_efficientnet_l2_ns_475
  • tf_efficientnet_lite0
  • tf_efficientnet_lite1
  • tf_efficientnet_lite2
  • tf_efficientnet_lite3
  • tf_efficientnet_lite4
  • tf_efficientnetv2_b0
  • tf_efficientnetv2_b1
  • tf_efficientnetv2_b2
  • tf_efficientnetv2_b3
  • tf_efficientnetv2_l
  • tf_efficientnetv2_l_in21ft1k
  • tf_efficientnetv2_l_in21k
  • tf_efficientnetv2_m
  • tf_efficientnetv2_m_in21ft1k
  • tf_efficientnetv2_m_in21k
  • tf_efficientnetv2_s
  • tf_efficientnetv2_s_in21ft1k
  • tf_efficientnetv2_s_in21k
  • tf_efficientnetv2_xl_in21ft1k
  • tf_efficientnetv2_xl_in21k
  • tf_inception_v3
  • tf_mixnet_l
  • tf_mixnet_m
  • tf_mixnet_s
  • tf_mobilenetv3_large_075
  • tf_mobilenetv3_large_100
  • tf_mobilenetv3_large_minimal_100
  • tf_mobilenetv3_small_075
  • tf_mobilenetv3_small_100
  • tf_mobilenetv3_small_minimal_100
  • tnt_b_patch16_224
  • tnt_s_patch16_224
  • tresnet_l
  • tresnet_l_448
  • tresnet_m
  • tresnet_m_448
  • tresnet_m_miil_in21k
  • tresnet_xl
  • tresnet_xl_448
  • tv_densenet121
  • tv_resnet101
  • tv_resnet152
  • tv_resnet34
  • tv_resnet50
  • tv_resnext50_32x4d
  • twins_pcpvt_base
  • twins_pcpvt_large
  • twins_pcpvt_small
  • twins_svt_base
  • twins_svt_large
  • twins_svt_small
  • vgg11
  • vgg11_bn
  • vgg13
  • vgg13_bn
  • vgg16
  • vgg16_bn
  • vgg19
  • vgg19_bn
  • visformer_small
  • visformer_tiny
  • vit_base_patch16_224
  • vit_base_patch16_224_in21k
  • vit_base_patch16_224_miil
  • vit_base_patch16_224_miil_in21k
  • vit_base_patch16_384
  • vit_base_patch16_sam_224
  • vit_base_patch32_224
  • vit_base_patch32_224_in21k
  • vit_base_patch32_384
  • vit_base_patch32_sam_224
  • vit_base_r26_s32_224
  • vit_base_r50_s16_224
  • vit_base_r50_s16_224_in21k
  • vit_base_r50_s16_384
  • vit_base_resnet26d_224
  • vit_base_resnet50_224_in21k
  • vit_base_resnet50_384
  • vit_base_resnet50d_224
  • vit_huge_patch14_224_in21k
  • vit_large_patch16_224
  • vit_large_patch16_224_in21k
  • vit_large_patch16_384
  • vit_large_patch32_224
  • vit_large_patch32_224_in21k
  • vit_large_patch32_384
  • vit_large_r50_s32_224
  • vit_large_r50_s32_224_in21k
  • vit_large_r50_s32_384
  • vit_small_patch16_224
  • vit_small_patch16_224_in21k
  • vit_small_patch16_384
  • vit_small_patch32_224
  • vit_small_patch32_224_in21k
  • vit_small_patch32_384
  • vit_small_r26_s32_224
  • vit_small_r26_s32_224_in21k
  • vit_small_r26_s32_384
  • vit_small_resnet26d_224
  • vit_small_resnet50d_s16_224
  • vit_tiny_patch16_224
  • vit_tiny_patch16_224_in21k
  • vit_tiny_patch16_384
  • vit_tiny_r_s16_p8_224
  • vit_tiny_r_s16_p8_224_in21k
  • vit_tiny_r_s16_p8_384
  • vovnet39a
  • vovnet57a
  • wide_resnet101_2
  • wide_resnet50_2
  • xception
  • xception41
  • xception65
  • xception71
  • xcit_large_24_p16_224
  • xcit_large_24_p16_224_dist
  • xcit_large_24_p16_384_dist
  • xcit_large_24_p8_224
  • xcit_large_24_p8_224_dist
  • xcit_large_24_p8_384_dist
  • xcit_medium_24_p16_224
  • xcit_medium_24_p16_224_dist
  • xcit_medium_24_p16_384_dist
  • xcit_medium_24_p8_224
  • xcit_medium_24_p8_224_dist
  • xcit_medium_24_p8_384_dist
  • xcit_nano_12_p16_224
  • xcit_nano_12_p16_224_dist
  • xcit_nano_12_p16_384_dist
  • xcit_nano_12_p8_224
  • xcit_nano_12_p8_224_dist
  • xcit_nano_12_p8_384_dist
  • xcit_small_12_p16_224
  • xcit_small_12_p16_224_dist
  • xcit_small_12_p16_384_dist
  • xcit_small_12_p8_224
  • xcit_small_12_p8_224_dist
  • xcit_small_12_p8_384_dist
  • xcit_small_24_p16_224
  • xcit_small_24_p16_224_dist
  • xcit_small_24_p16_384_dist
  • xcit_small_24_p8_224
  • xcit_small_24_p8_224_dist
  • xcit_small_24_p8_384_dist
  • xcit_tiny_12_p16_224
  • xcit_tiny_12_p16_224_dist
  • xcit_tiny_12_p16_384_dist
  • xcit_tiny_12_p8_224
  • xcit_tiny_12_p8_224_dist
  • xcit_tiny_12_p8_384_dist
  • xcit_tiny_24_p16_224
  • xcit_tiny_24_p16_224_dist
  • xcit_tiny_24_p16_384_dist
  • xcit_tiny_24_p8_224
  • xcit_tiny_24_p8_224_dist
  • xcit_tiny_24_p8_384_dist