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

Set XLA example time to 500s

Browse Source
This commit is contained in:
Lysandre
2020-10-15 12:34:29 +02:00
parent 2dba7d5702
commit 2485b8b0ac
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/test_xla_examples.py
View File

@@ -80,7 +80,7 @@ class TorchXLAExamplesTests(unittest.TestCase):
self.assertGreaterEqual(value, 0.70) self.assertGreaterEqual(value, 0.70)
# Assert that the script takes less than 300 seconds to make sure it doesn't hang. # Assert that the script takes less than 300 seconds to make sure it doesn't hang.
self.assertLess(end - start, 300) self.assertLess(end - start, 500)
def test_trainer_tpu(self): def test_trainer_tpu(self):
import xla_spawn import xla_spawn