train_many2one.py 文件源码

def decode(debug=True):
  """ Decode file """
  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)

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

    dev_set = load_dev_data()
    eval_batch_size = FLAGS.batch_size

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