mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
90b4e130d6
Prune models from TorchInductor dashboard to reduce ci cost. This PR prunes torchbench models according to the [doc](https://docs.google.com/document/d/1nLPNNAU-_M9Clx9FMrJ1ycdPxe-xRA54olPnsFzdpoU/edit?tab=t.0), which removes timm and huggingface models from torchbench. Pull Request resolved: https://github.com/pytorch/pytorch/pull/164816 Approved by: https://github.com/anijain2305, https://github.com/seemethere, https://github.com/huydhn, https://github.com/malfet
18 lines
295 B
Plaintext
18 lines
295 B
Plaintext
BERT_pytorch,128
|
|
Background_Matting, 16
|
|
LearningToPaint,1024
|
|
alexnet,1024
|
|
dcgan,1024
|
|
densenet121,64
|
|
mnasnet1_0,256
|
|
mobilenet_v2,128
|
|
mobilenet_v3_large,256
|
|
nvidia_deeprecommender,1024
|
|
pytorch_unet,8
|
|
resnet18,512
|
|
resnet50,128
|
|
resnext50_32x4d,128
|
|
shufflenet_v2_x1_0,512
|
|
squeezenet1_1,512
|
|
vgg16,128
|