From 9a3453846b40dc58d549d6ab5cd9cd76d4be74be Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 27 Jun 2022 14:36:11 +0200 Subject: [PATCH] fix (#17890) Co-authored-by: ydshieh --- tests/models/layoutlmv2/test_modeling_layoutlmv2.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/tests/models/layoutlmv2/test_modeling_layoutlmv2.py b/tests/models/layoutlmv2/test_modeling_layoutlmv2.py index bfcd729df1..35eef14d2b 100644 --- a/tests/models/layoutlmv2/test_modeling_layoutlmv2.py +++ b/tests/models/layoutlmv2/test_modeling_layoutlmv2.py @@ -20,7 +20,7 @@ import random import tempfile import unittest -from transformers.testing_utils import require_detectron2, require_torch, slow, torch_device +from transformers.testing_utils import require_detectron2, require_torch, require_torch_multi_gpu, slow, torch_device from transformers.utils import is_detectron2_available, is_torch_available from ...test_configuration_common import ConfigTester @@ -285,6 +285,16 @@ class LayoutLMv2ModelTest(ModelTesterMixin, unittest.TestCase): config_and_inputs = self.model_tester.prepare_config_and_inputs() self.model_tester.create_and_check_model(*config_and_inputs) + @require_torch_multi_gpu + @unittest.skip( + reason=( + "LayoutLMV2 and its dependency `detectron2` have some layers using `add_module` which doesn't work well" + " with `nn.DataParallel`" + ) + ) + def test_multi_gpu_data_parallel_forward(self): + pass + def test_model_various_embeddings(self): config_and_inputs = self.model_tester.prepare_config_and_inputs() for type in ["absolute", "relative_key", "relative_key_query"]: