xref: /aosp_15_r20/external/pytorch/benchmarks/dynamo/all_torchbench_models_list.txt (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1BERT_pytorch
2Background_Matting
3DALLE2_pytorch
4LearningToPaint
5Super_SloMo
6alexnet
7attention_is_all_you_need_pytorch
8dcgan
9demucs
10densenet121
11detectron2_fasterrcnn_r_101_c4
12detectron2_fasterrcnn_r_101_dc5
13detectron2_fasterrcnn_r_101_fpn
14detectron2_fasterrcnn_r_50_c4
15detectron2_fasterrcnn_r_50_dc5
16detectron2_fasterrcnn_r_50_fpn
17detectron2_fcos_r_50_fpn
18detectron2_maskrcnn
19detectron2_maskrcnn_r_101_c4
20detectron2_maskrcnn_r_101_fpn
21detectron2_maskrcnn_r_50_c4
22detectron2_maskrcnn_r_50_fpn
23dlrm
24drq
25fambench_dlrm
26fambench_xlmr
27fastNLP_Bert
28hf_Albert
29hf_Bart
30hf_Bert
31hf_BigBird
32hf_DistilBert
33hf_GPT2
34hf_Longformer
35hf_Reformer
36hf_T5
37maml
38maml_omniglot
39mnasnet1_0
40mobilenet_v2
41mobilenet_v2_quantized_qat
42mobilenet_v3_large
43moco
44nvidia_deeprecommender
45opacus_cifar10
46pplbench_beanmachine
47pyhpc_equation_of_state
48pyhpc_isoneutral_mixing
49pyhpc_turbulent_kinetic_energy
50pytorch_CycleGAN_and_pix2pix
51pytorch_stargan
52pytorch_struct
53pytorch_unet
54resnet18
55resnet50
56resnet50_quantized_qat
57resnext50_32x4d
58shufflenet_v2_x1_0
59soft_actor_critic
60speech_transformer
61squeezenet1_1
62tacotron2
63timm_efficientdet
64timm_efficientnet
65timm_nfnet
66timm_regnet
67timm_resnest
68timm_vision_transformer
69timm_vovnet
70tts_angular
71vgg16
72vision_maskrcnn
73yolov3