Esempi in Python per init_from_checkpoint

Esempio n. 1

File: reinforce_model.py Progetto: google-research/language

 def scaffold_fn():
     """Init op run on host."""
     checkpoint = params["base_model"]
     if params["warm_start_path"]:
         checkpoint = params["warm_start_path"]
     if checkpoint:
         checkpoint_utils.init_from_checkpoint(checkpoint)
     return tf.train.Scaffold()

Esempio n. 2

File: infer_wsp_captions.py Progetto: xiaoqiangzhang203/language

 def scaffold_fn():
     """Init op run on host."""
     checkpoint_utils.init_from_checkpoint(params["checkpoint"])
     return tf.train.Scaffold()

Esempio n. 3

 def scaffold_fn():
     """Init op run on host."""
     checkpoint = params.get("warm_start_path")
     if checkpoint:
         checkpoint_utils.init_from_checkpoint(checkpoint)
     return tf.train.Scaffold()