def get_checkpoint_model(self, version): """Read checkpoint using model version""" file = self.get_checkpoint_path(version) try: return load_from_checkpoint_file(file) except Exception: raise RuntimeError("Failed to read model checkpoint from file " + file)
def _init_model(self, checkpoint_filename_for_init, init_var): if checkpoint_filename_for_init: pb_model = load_from_checkpoint_file(checkpoint_filename_for_init) self._version = pb_model.version self._init_model_from_tensor_dict(pb_model.param) elif init_var: self._init_model_from_var_list(init_var) else: logger.info("Model is not intialized. It will be " "initialized by the first update from " "the worker.")