Skip to content

GitLab

Unverified Commit 52ecad8d authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Exclude MViT from cuda tests (#6573)

parent 3894c353
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/test_models.py
View file @ 52ecad8d
...@@ -348,8 +348,8 @@ for m in slow_models: ...@@ -348,8 +348,8 @@ for m in slow_models:
skipped_big_models = { skipped_big_models = {
"vit_h_14": {("Windows", "cpu"), ("Windows", "cuda")}, "vit_h_14": {("Windows", "cpu"), ("Windows", "cuda")},
"regnet_y_128gf": {("Windows", "cpu"), ("Windows", "cuda")}, "regnet_y_128gf": {("Windows", "cpu"), ("Windows", "cuda")},
"mvit_v1_b": {("Windows", "cuda")}, "mvit_v1_b": {("Windows", "cuda"), ("Linux", "cuda")},
"mvit_v2_s": {("Windows", "cuda")}, "mvit_v2_s": {("Windows", "cuda"), ("Linux", "cuda")},
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment