decode_one.py 文件源码

def decode(debug=True):
  """ Decode file sentence-by-sentence  """
  with tf.Session(config=tf.ConfigProto(intra_op_parallelism_threads=NUM_THREADS)) as sess:
    # Create model and load parameters.
    with tf.variable_scope("model", reuse=None):
      model_dev, steps_done = create_model(sess, forward_only=True, dropout=False, model_path=model_path)

    if debug:
      for v in tf.all_variables(): print(v.name, v.get_shape())

    eval_batch_size = 64
    start_time = time.time()
    do_evalb(model_dev, sess, dev_set, eval_batch_size)
    time_elapsed = time.time() - start_time
    print("Batched evalb time: ", time_elapsed)

#    start_time = time.time()
#    write_decode(model_dev, sess, dev_set, eval_batch_size) 
#    time_elapsed = time.time() - start_time
#    print("Decoding all dev time: ", time_elapsed)