OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Add timm_efficientnet to flaky models after cuda 12.6 update in CI/CD (#148788)

Browse Source 
After https://github.com/pytorch/pytorch/pull/148612
This model have become flaky

Tracking this regression in an issue : https://github.com/pytorch/pytorch/issues/148699

Pull Request resolved: https://github.com/pytorch/pytorch/pull/148788
Approved by: https://github.com/izaitsevfb, https://github.com/malfet
This commit is contained in:
atalman 2025-03-10 13:40:41 +00:00 committed by PyTorch MergeBot
parent 68c12ecfe2
commit 2068235c0a
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
benchmarks/dynamo/check_accuracy.py
View File

@ -12,6 +12,7 @@ flaky_models = {
"yolov3", "yolov3",
"gluon_inception_v3", "gluon_inception_v3",
"detectron2_maskrcnn_r_101_c4", "detectron2_maskrcnn_r_101_c4",
"timm_efficientnet", # see https://github.com/pytorch/pytorch/issues/148699
"XGLMForCausalLM", # discovered in https://github.com/pytorch/pytorch/pull/128148 "XGLMForCausalLM", # discovered in https://github.com/pytorch/pytorch/pull/128148
} }