diff --git a/tests/test_highresnet.py b/tests/test_highresnet.py index 83248ad85f..61af529b63 100644 --- a/tests/test_highresnet.py +++ b/tests/test_highresnet.py @@ -53,7 +53,7 @@ def test_shape(self, input_param, input_shape, expected_shape): result = net.forward(torch.randn(input_shape).to(device)) self.assertEqual(result.shape, expected_shape) - @TimedCall(seconds=400, force_quit=True) + @TimedCall(seconds=800, force_quit=True) def test_script(self): input_param, input_shape, expected_shape = TEST_CASE_1 net = HighResNet(**input_param) diff --git a/tests/test_integration_determinism.py b/tests/test_integration_determinism.py index 4947610484..e077420420 100644 --- a/tests/test_integration_determinism.py +++ b/tests/test_integration_determinism.py @@ -75,7 +75,7 @@ def setUp(self): def tearDown(self): set_determinism(seed=None) - @TimedCall(seconds=30) + @TimedCall(seconds=150) def test_training(self): set_determinism(seed=0) loss, step = run_test(device=self.device)