Fix flaky non-contig test

fxia22 · Feb 17, 2017 · 7117a90 · 7117a90
1 parent 1bdc281
commit 7117a90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):