def demo(): 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.demo(sess, model, FLAGS)
def demo(buckets=None): 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.demo(sess, model, FLAGS)