Python CheckpointSaver.get_version_from_checkpoint Examples

Example #1

File: save_utils_test.py Project: zerocurve/elasticdl

 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)

Example #2

File: task_manager.py Project: Terry1504/elasticdl

    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)

Example #3

    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)