Skip to content
This repository was archived by the owner on Aug 1, 2025. It is now read-only.

Commit 78056e2

Browse files
authored
Skip tacotron2 and unskip vision_maskrcnn in torchbench.py (#102)
1 parent c790189 commit 78056e2

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

torchbench.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,10 @@
5555
SKIP = {
5656
# non-deterministic output / cant check correctness
5757
"pyhpc_turbulent_kinetic_energy",
58-
# CUDA torchvision::nms build issues on AWS cluser
58+
# https://github.com/facebookresearch/torchdynamo/issues/82
59+
"tacotron2",
60+
# https://github.com/facebookresearch/torchdynamo/issues/101
5961
"detectron2_maskrcnn",
60-
"vision_maskrcnn",
6162
}
6263

6364
# Additional models that are skipped in training

0 commit comments

Comments
 (0)