diff options
author | Todd Wang <toddw@google.com> | 2018-10-08 09:42:50 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-08 09:48:19 -0700 |
commit | 411b9baa39636030181fdff15d2e985824b03d61 (patch) | |
tree | 93b2a6cae1e8fd4cde251d4bbd45af8fa323f97c /tensorflow/contrib | |
parent | 5f308cb408eb46ec9af0546be6b9ae1d5166b185 (diff) |
Reduce tolerances for rmsprop_test float16, to fix OSS builds.
PiperOrigin-RevId: 216200439
Diffstat (limited to 'tensorflow/contrib')
-rw-r--r-- | tensorflow/contrib/optimizer_v2/rmsprop_test.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/contrib/optimizer_v2/rmsprop_test.py b/tensorflow/contrib/optimizer_v2/rmsprop_test.py index 44301ffe9e..83f5971039 100644 --- a/tensorflow/contrib/optimizer_v2/rmsprop_test.py +++ b/tensorflow/contrib/optimizer_v2/rmsprop_test.py @@ -157,8 +157,11 @@ class RMSPropOptimizerTest(test.TestCase, parameterized.TestCase): self.assertAllCloseAccordingToType(rms1_np, rms1.eval()) self.assertAllCloseAccordingToType(mom0_np, mom0.eval()) self.assertAllCloseAccordingToType(mom1_np, mom1.eval()) - self.assertAllCloseAccordingToType(var0_np, var0.eval()) - self.assertAllCloseAccordingToType(var1_np, var1.eval()) + # TODO(b/117393988): Reduce tolerances for float16. + self.assertAllCloseAccordingToType( + var0_np, var0.eval(), half_rtol=3e-3, half_atol=3e-3) + self.assertAllCloseAccordingToType( + var1_np, var1.eval(), half_rtol=3e-3, half_atol=3e-3) @parameterized.parameters([dtypes.float32, dtypes.float64]) def testMinimizeSparseResourceVariable(self, dtype): |