Increase timeout of tests (#2711)

Signed-off-by: Mohammad Adil <[email protected]>

Author: madil90

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)

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)