Searched refs:load_checkpoint (Results 1 – 6 of 6) sorted by relevance
768 load_checkpoint = trackable_utils.Checkpoint(v=variables_lib.Variable(0.))771 status = load_checkpoint.read(checkpoint_prefix)775 self.assertEqual(1., self.evaluate(load_checkpoint.v))779 status = load_checkpoint.read(checkpoint_prefix)783 self.assertEqual(3., self.evaluate(load_checkpoint.v))792 load_checkpoint = trackable_utils.Checkpoint(795 load_checkpoint.restore(save_path)796 self.assertAllClose(self.evaluate(load_checkpoint.a), [0, 1])797 self.assertAllClose(self.evaluate(load_checkpoint.b), {"a": 2, "b": 3})821 load_checkpoint = trackable_utils.Checkpoint(new_model)[all …]
46 def load_checkpoint(ckpt_dir_or_file): function84 reader = load_checkpoint(ckpt_dir_or_file)112 reader = load_checkpoint(ckpt_dir_or_file)312 reader = load_checkpoint(ckpt_dir_or_file)
78 from tensorflow.python.training.checkpoint_utils import load_checkpoint
52 reader = checkpoint_utils.load_checkpoint(ckpt_dir_or_file)
96 name: "load_checkpoint"
372 name: "load_checkpoint"