Python load_from_checkpoint_file Examples

Example #1

File: checkpoint_service.py Project: zhuli19901106/elasticdl

 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)

Example #2

File: servicer.py Project: gavinljj/elasticdl

 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.")