diff options
| author |  Amit Patankar <amitpatankar@google.com> | 2018-06-05 11:11:16 -0700 |
|---|---|---|
| committer | Amit Patankar <amitpatankar@google.com> | 2018-06-05 11:11:16 -0700 |
| commit | fdc085f021f98e7f4cba44e716f4f85cb9704447 (patch) | |
| tree | 4cdccd00ae075fdff548546d4c88c52ea700b85d | |
| parent | 83543deedb68fef61ea7e709de3f462a1edd13ce (diff) | |
Fixing the adamax_test rtol to be more lenient.
| -rw-r--r-- | tensorflow/contrib/opt/python/training/adamax_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/opt/python/training/adamax_test.py b/tensorflow/contrib/opt/python/training/adamax_test.py index 21bf3f5313..a059aae130 100644 --- a/tensorflow/contrib/opt/python/training/adamax_test.py +++ b/tensorflow/contrib/opt/python/training/adamax_test.py @@ -224,8 +224,8 @@ class AdaMaxOptimizerTest(test.TestCase): var1_np, m1, v1 = adamax_update_numpy(var1_np, grads1_np, t, m1, v1) # Validate updated params - self.assertAllCloseAccordingToType(var0_np, self.evaluate(var0)) - self.assertAllCloseAccordingToType(var1_np, self.evaluate(var1)) + self.assertAllCloseAccordingToType(var0_np, self.evaluate(var0), rtol=1e-2) + self.assertAllCloseAccordingToType(var1_np, self.evaluate(var1), rtol=1e-2) if use_resource: self.assertEqual("var0_%d/AdaMax:0" % (i,), opt.get_slot(var=var0, name="m").name) |