From 5972cabc39ef968a625efd4ca55936101a6a035c Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Tue, 13 Sep 2022 11:47:15 +0100 Subject: [PATCH] Exclude MViT from cuda tests --- test/test_models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_models.py b/test/test_models.py index 2629c3c4dc1..29b57c60cca 100644 --- a/test/test_models.py +++ b/test/test_models.py @@ -348,8 +348,8 @@ def _check_input_backprop(model, inputs): skipped_big_models = { "vit_h_14": {("Windows", "cpu"), ("Windows", "cuda")}, "regnet_y_128gf": {("Windows", "cpu"), ("Windows", "cuda")}, - "mvit_v1_b": {("Windows", "cuda")}, - "mvit_v2_s": {("Windows", "cuda")}, + "mvit_v1_b": {("Windows", "cuda"), ("Linux", "cuda")}, + "mvit_v2_s": {("Windows", "cuda"), ("Linux", "cuda")}, }