def decode(data_set, buckets=None, verbose=True): with tf.Session(config=tf.ConfigProto(allow_soft_placement=True, log_device_placement=FLAGS.log_device_placement)) as sess: # Initialize model parameters. model = define_model(sess, forward_only=True, buckets=buckets) decode_tools.decode_set(sess, model, data_set, 3, FLAGS, verbose) return model
def decode(data_set, verbose=True): with tf.Session(config=tf.ConfigProto( allow_soft_placement=True, log_device_placement=FLAGS.log_device_placement)) as sess: # Create model and load parameters. model = create_model(sess, forward_only=True) decode_tools.decode_set(sess, model, data_set, 3, FLAGS, verbose) return model.model_dir, model.decode_sig