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

Skip some detectron2_maskrcnn models with KeyError _ignore_torch_cuda_oom (#99599)

Browse Source 
These tests are failing in trunk 233cc34d3b with `KeyError: '_ignore_torch_cuda_oom'`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/99599
Approved by: https://github.com/malfet
This commit is contained in:
Huy Do 2023-04-20 18:11:35 +00:00 committed by PyTorch MergeBot
parent 7c3fa5c70d
commit 5315317b7b
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
benchmarks/dynamo/common.py
View File

@ -84,6 +84,11 @@ CI_SKIP[CI("eager", training=False)] = [
"torchrec_dlrm",
# Huggingface
"DebertaV2ForQuestionAnswering", # OOM
# KeyError: '_ignore_torch_cuda_oom'
"detectron2_maskrcnn_r_101_c4",
"detectron2_maskrcnn_r_101_fpn",
"detectron2_maskrcnn_r_50_c4",
"detectron2_maskrcnn_r_50_fpn",
]
CI_SKIP[CI("eager", training=True)] = [