[fbsync] Skip big models on both cpu and gpu test to fix CI(#6197)

YosuaMichael · facebook-github-bot · commit 135f6ec77f7f · 2022-06-27T13:55:03.000-07:00
Reviewed By: NicolasHug

Differential Revision: D37450354

fbshipit-source-id: 997deeced50eea5a3df00ea74f55b6ceadf21caa
diff --git a/test/test_models.py b/test/test_models.py
@@ -603,7 +603,7 @@ def test_classification_model(model_fn, dev):
         "input_shape": (1, 3, 224, 224),
     }
     model_name = model_fn.__name__
-    if dev == "cuda" and SKIP_BIG_MODEL and model_name in skipped_big_models:
+    if SKIP_BIG_MODEL and model_name in skipped_big_models:
         pytest.skip("Skipped to reduce memory usage. Set env var SKIP_BIG_MODEL=0 to enable test for this model")
     kwargs = {**defaults, **_model_params.get(model_name, {})}
     num_classes = kwargs.get("num_classes")

Original file line number	Diff line number	Diff line change
`@@ -603,7 +603,7 @@ def test_classification_model(model_fn, dev):`
`603`	`603`	`"input_shape": (1, 3, 224, 224),`
`604`	`604`	`}`
`605`	`605`	`model_name = model_fn.__name__`
`606`		`- if dev == "cuda" and SKIP_BIG_MODEL and model_name in skipped_big_models:`
	`606`	`+ if SKIP_BIG_MODEL and model_name in skipped_big_models:`
`607`	`607`	`pytest.skip("Skipped to reduce memory usage. Set env var SKIP_BIG_MODEL=0 to enable test for this model")`
`608`	`608`	`kwargs = {defaults, _model_params.get(model_name, {})}`
`609`	`609`	`num_classes = kwargs.get("num_classes")`