Set XLA example time to 500s

2020-10-15 12:34:29 +02:00
parent 2dba7d5702
commit 2485b8b0ac
1 changed files with 1 additions and 1 deletions
--- a/examples/test_xla_examples.py
+++ b/examples/test_xla_examples.py
@@ -80,7 +80,7 @@ class TorchXLAExamplesTests(unittest.TestCase):
                self.assertGreaterEqual(value, 0.70)

            # 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):
        import xla_spawn