SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

skip test_multi_gpu_data_parallel_forward for MaskFormerSwinModelTest (#20688)

Browse Source 
Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar
2022-12-09 11:10:00 +01:00
committed by GitHub
parent 6a062a3ed9
commit 704027f0ef
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/models/maskformer/test_modeling_maskformer_swin.py
View File

@@ -20,7 +20,7 @@ import unittest
from typing import Dict, List, Tuple from typing import Dict, List, Tuple
from transformers import MaskFormerSwinConfig from transformers import MaskFormerSwinConfig
from transformers.testing_utils import require_torch, torch_device from transformers.testing_utils import require_torch, require_torch_multi_gpu, torch_device
from transformers.utils import is_torch_available from transformers.utils import is_torch_available
from ...test_configuration_common import ConfigTester from ...test_configuration_common import ConfigTester
@@ -182,6 +182,16 @@ class MaskFormerSwinModelTest(ModelTesterMixin, unittest.TestCase):
self.model_tester = MaskFormerSwinModelTester(self) self.model_tester = MaskFormerSwinModelTester(self)
self.config_tester = ConfigTester(self, config_class=MaskFormerSwinConfig, embed_dim=37) self.config_tester = ConfigTester(self, config_class=MaskFormerSwinConfig, embed_dim=37)
@require_torch_multi_gpu
@unittest.skip(
reason=(
"`MaskFormerSwinModel` outputs `hidden_states_spatial_dimensions` which doesn't work well with"
" `nn.DataParallel`"
)
)
def test_multi_gpu_data_parallel_forward(self):
pass
def test_config(self): def test_config(self):
self.create_and_test_config_common_properties() self.create_and_test_config_common_properties()
self.config_tester.create_and_test_config_to_json_string() self.config_tester.create_and_test_config_to_json_string()