From 2485b8b0ac99ff40d48d6007a40436c8da26b53d Mon Sep 17 00:00:00 2001 From: Lysandre Date: Thu, 15 Oct 2020 12:34:29 +0200 Subject: [PATCH] Set XLA example time to 500s --- examples/test_xla_examples.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/test_xla_examples.py b/examples/test_xla_examples.py index c5539eee1f..cd34c5b53b 100644 --- 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