model_helper.py 文件源码

def create_or_load_model(model, model_dir, session, out_dir, name):
  """Create translation model and initialize or load parameters in session."""
  start_time = time.time()
  latest_ckpt = tf.train.latest_checkpoint(model_dir)
  if latest_ckpt:
    model.saver.restore(session, latest_ckpt)
    utils.print_out(
        "  loaded %s model parameters from %s, time %.2fs" %
        (name, latest_ckpt, time.time() - start_time))
  else:
    utils.print_out("  created %s model with fresh parameters, time %.2fs." %
                    (name, time.time() - start_time))
    session.run(tf.global_variables_initializer())

  session.run(tf.tables_initializer())

  global_step = model.global_step.eval(session=session)
  return model, global_step