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

[testing] rename skip targets + docs (#7863)

Browse Source 
* rename skip targets + docs

* fix quotes

* style

* Apply suggestions from code review

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

* small improvements

* fix

Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
This commit is contained in:
Stas Bekman
2020-10-20 01:39:13 -07:00
committed by GitHub
parent c912ba5f69
commit 3e31e7f956
9 changed files with 52 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
templates/adding_a_new_model/tests/test_modeling_xxx.py
View File

@@ -17,7 +17,7 @@
import unittest
from transformers import is_torch_available
from transformers.testing_utils import require_torch, require_torch_and_cuda, slow, torch_device
from transformers.testing_utils import require_torch, require_torch_gpu, slow, torch_device
from .test_configuration_common import ConfigTester
from .test_modeling_common import ModelTesterMixin, ids_tensor
@@ -302,6 +302,6 @@ class XxxModelTest(ModelTesterMixin, unittest.TestCase):
"""Test loss/gradients same as reference implementation, for example."""
pass
@require_torch_and_cuda
@require_torch_gpu
def test_large_inputs_in_fp16_dont_cause_overflow(self):
pass