Searched refs:load_checkpoint (Results 1 – 11 of 11) sorted by relevance
786 load_checkpoint = trackable_utils.Checkpoint(v=variables_lib.Variable(0.))789 status = load_checkpoint.read(checkpoint_prefix)793 self.assertEqual(1., self.evaluate(load_checkpoint.v))797 status = load_checkpoint.read(checkpoint_prefix)801 self.assertEqual(3., self.evaluate(load_checkpoint.v))819 load_checkpoint = trackable_utils.Checkpoint(v=variables_lib.Variable(0.))822 status = load_checkpoint.read(checkpoint_prefix)826 self.assertEqual(1., self.evaluate(load_checkpoint.v))830 status = load_checkpoint.read(checkpoint_prefix)834 self.assertEqual(3., self.evaluate(load_checkpoint.v))[all …]
61 legacy_reader = checkpoint_utils.load_checkpoint(63 reader = checkpoint_utils.load_checkpoint(checkpoint_path)
47 def load_checkpoint(ckpt_dir_or_file): function85 reader = load_checkpoint(ckpt_dir_or_file)113 reader = load_checkpoint(ckpt_dir_or_file)357 reader = load_checkpoint(ckpt_dir_or_file)
73 from tensorflow.python.training.checkpoint_utils import load_checkpoint
124 reader = checkpoint_utils.load_checkpoint(checkpoint_dir)
25 def load_checkpoint(cfg: Any) -> Dict[str, Any]: # pyre-ignore[2] function35 checkpoint_dict = checkpointer.load_checkpoint()
14 load_checkpoint,73 checkpoint_dict = load_checkpoint(cfg=cfg)
17 3. Make sure you have a ``load_checkpoint(path)`` and35 state = load_checkpoint(args.checkpoint_path)
104 name: "load_checkpoint"
61 sd = checkpointer.load_checkpoint()
372 name: "load_checkpoint"