experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add test for checkpoint released before status is checked.

Browse Source 
PiperOrigin-RevId: 256244578
This commit is contained in:
Tom Hennigan 2019-07-02 14:52:39 -07:00 committed by TensorFlower Gardener
parent 54a771061d
commit e47ee3f5d2
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/python/training/tracking/util_test.py
View File

@ -397,6 +397,18 @@ class CheckpointingTests(parameterized.TestCase, test.TestCase):
self.assertEqual(42., self.evaluate(v.non_dep_variable))
self.assertEqual(42., self.evaluate(v.mirrored))
@test_util.run_in_graph_and_eager_modes
def testAssertConsumedNoCheckpoint(self):
prefix = os.path.join(self.get_temp_dir(), "ckpt")
v = variable_scope.get_variable(name="v", initializer=0.)
self.evaluate(v.initializer)
ckpt = trackable_utils.Checkpoint(v=v)
self.evaluate(trackable_utils.gather_initializers(ckpt))
save_path = ckpt.save(file_prefix=prefix)
status = ckpt.restore(save_path=save_path)
del ckpt
status.assert_consumed()
@test_util.run_in_graph_and_eager_modes
def testSaveRestore(self):
model = MyModel()