Python disable_check_numerics Examples

Example #1

File: math_grad_test.py Project: MFChunga/poo

 def test_zero_grad_tape(self):
     try:
         check_numerics_callback.enable_check_numerics()
         x = constant_op.constant([-1, 0., 1.])
         with backprop.GradientTape() as tape:
             tape.watch(x)
             g = tape.gradient(math_ops.pow(x, 2), x)
         g = self.evaluate(g)
         self.assertAllClose([-2., 0., 2.], g)
     finally:
         check_numerics_callback.disable_check_numerics()

Example #2

File: check_numerics_callback_test.py Project: neuroph12/tensorflow-cplusplus-learn

 def tearDown(self):
     check_numerics_callback.disable_check_numerics()
     super(CheckNumericsCallbackTest, self).tearDown()

Example #3

File: check_numerics_callback_test.py Project: neuroph12/tensorflow-cplusplus-learn

 def testCallingDisableCheckNumericsWithoutEnablingFirstIsTolerated(self):
     check_numerics_callback.disable_check_numerics()

Example #4

 def tearDown(self):
     if os.path.isdir(self.dump_root):
         shutil.rmtree(self.dump_root, ignore_errors=True)
     check_numerics_callback.disable_check_numerics()
     dumping_callback.disable_dump_debug_info()
     super(DumpingCallbackTestBase, self).tearDown()