From 7117a9012ee33f812334c21ae389942986d24a32 Mon Sep 17 00:00:00 2001 From: Adam Paszke Date: Thu, 16 Feb 2017 09:34:31 -0800 Subject: [PATCH] Fix flaky non-contig test --- test/common_nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/common_nn.py b/test/common_nn.py index 1fa67e9cb48cb..1bad987eb9af1 100644 --- a/test/common_nn.py +++ b/test/common_nn.py @@ -573,7 +573,7 @@ def test_noncontig(self, test_case, module, input): grad = test_case._backward(module, i, out, go) test_case.assertEqual(out, output) - test_case.assertEqual(grad, d_input) + test_case.assertEqual(grad, d_input, 1e-4) test_case.assertEqual(test_case._get_parameters(module)[1], d_param) def test_cuda(self, test_case):