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