def testGetVersionFromCheckpoint(self): with tempfile.TemporaryDirectory() as tempdir: self.params.version = 100 ckpt_dir = save_variables_to_checkpoint(tempdir, self.params) ckpt_version_dir = os.path.join(ckpt_dir, "version-100") model_version = CheckpointSaver.get_version_from_checkpoint( ckpt_version_dir) self.assertTrue(model_version, 100)
def _set_completed_steps_by_checkpoint(self, checkpoint_dir_for_init): if not checkpoint_dir_for_init: return if not CheckpointSaver.check_checkpoint_valid(checkpoint_dir_for_init): raise ValueError("Invalid checkpoint directory {}".format( checkpoint_dir_for_init)) self._completed_steps = CheckpointSaver.get_version_from_checkpoint( checkpoint_dir_for_init)
def _set_completed_steps_by_checkpoint(self, checkpoint_dir_for_init): if not checkpoint_dir_for_init: return if not CheckpointSaver.check_checkpoint_valid(checkpoint_dir_for_init): raise ValueError("Invalid checkpoint directory {}".format( checkpoint_dir_for_init)) model_verion = CheckpointSaver.get_version_from_checkpoint( checkpoint_dir_for_init) for callback in self.callbacks_list.callbacks: if isinstance(callback, MaxStepsStopping): callback.set_completed_steps(model_verion)