improve test robustness for Normalize (pytorch#1518)

Summary: Pull Request resolved: pytorch#1518 see title. This bumps the tolerances. This was flaky locally. Differential Revision: https://internalfb.com/D41482368 fbshipit-source-id: a6dfe2675a2ce078e2c3a6151176d4edeacf08d3
sdaulton · Feb 6, 2023 · 4fc59a4 · 4fc59a4
1 parent 715df0e
commit 4fc59a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/models/transforms/test_input.py b/test/models/transforms/test_input.py
@@ -232,7 +232,7 @@ def test_normalize(self):
 
                 nlz.eval()
                 X_unnlzd = nlz.untransform(X_nlzd)
-                self.assertAllClose(X, X_unnlzd, atol=1e-4, rtol=1e-4)
+                self.assertAllClose(X, X_unnlzd, atol=1e-3, rtol=1e-3)
                 expected_bounds = torch.cat(
                     [X.min(dim=-2, keepdim=True)[0], X.max(dim=-2, keepdim=True)[0]],
                     dim=-2,