[fbsync] Skip r2plus1d_18 test on linux GPU Circle CI machines (#3768)

Reviewed By: datumbox

Differential Revision: D28473320

fbshipit-source-id: 8ecc3772e63f79343cce18e68ab5516f7e841710

Author: NicolasHug

.circleci/config.yml

@@ -648,7 +648,7 @@ jobs:
           command: docker run -t --gpus all -v $PWD:$PWD -w $PWD -e UPLOAD_CHANNEL -e CU_VERSION "${image_name}" .circleci/unittest/linux/scripts/install.sh
       - run:
           name: Run tests
-          command: docker run -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
+          command: docker run -e CIRCLECI -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
       - run:
           name: Post Process
           command: docker run -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/post_process.sh

.circleci/config.yml.in

@@ -648,7 +648,7 @@ jobs:
           command: docker run -t --gpus all -v $PWD:$PWD -w $PWD -e UPLOAD_CHANNEL -e CU_VERSION "${image_name}" .circleci/unittest/linux/scripts/install.sh
       - run:
           name: Run tests
-          command: docker run -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
+          command: docker run -e CIRCLECI -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/run_test.sh
       - run:
           name: Post Process
           command: docker run -t --gpus all -v $PWD:$PWD -w $PWD "${image_name}" .circleci/unittest/linux/scripts/post_process.sh

test/common_utils.py

@@ -23,6 +23,7 @@
 IS_PY39 = sys.version_info.major == 3 and sys.version_info.minor == 9
 PY39_SEGFAULT_SKIP_MSG = "Segmentation fault with Python 3.9, see https://github.com/pytorch/vision/issues/3367"
 PY39_SKIP = unittest.skipIf(IS_PY39, PY39_SEGFAULT_SKIP_MSG)
+IN_CIRCLE_CI = os.getenv("CIRCLECI", False) == 'true'
 
 
 @contextlib.contextmanager

test/test_models.py

@@ -1,4 +1,5 @@
-from common_utils import TestCase, map_nested_tensor_object, freeze_rng_state, set_rng_seed
+import sys
+from common_utils import TestCase, map_nested_tensor_object, freeze_rng_state, set_rng_seed, IN_CIRCLE_CI
 from collections import OrderedDict
 from itertools import product
 import functools
@@ -458,6 +459,9 @@ def test_detection_model_validation(model_name):
 @pytest.mark.parametrize('model_name', get_available_video_models())
 @pytest.mark.parametrize('dev', _devs)
 def test_video_model(model_name, dev):
+    if IN_CIRCLE_CI and 'cuda' in dev.type and model_name == 'r2plus1d_18' and sys.platform == 'linux':
+        # FIXME: Failure should fixed and test re-actived. See https://github.com/pytorch/vision/issues/3702
+        pytest.skip('r2plus1d_18 fails on CircleCI linux GPU machines.')
     ModelTester()._test_video_model(model_name, dev)