xref: /aosp_15_r20/external/pytorch/benchmarks/dynamo/torchbench_models_list.txt (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1BERT_pytorch,128
2Background_Matting, 16
3LearningToPaint,1024
4alexnet,1024
5dcgan,1024
6densenet121,64
7hf_Albert,32
8hf_Bart,16
9hf_Bert,16
10hf_GPT2,16
11hf_T5,4
12mnasnet1_0,256
13mobilenet_v2,128
14mobilenet_v3_large,256
15nvidia_deeprecommender,1024
16pytorch_unet,8
17resnet18,512
18resnet50,128
19resnext50_32x4d,128
20shufflenet_v2_x1_0,512
21squeezenet1_1,512
22timm_nfnet,256
23timm_efficientnet,128
24timm_regnet,128
25timm_resnest,256
26timm_vision_transformer,256
27timm_vovnet,128
28vgg16,128
29