コード例 #1
0
ファイル: translate.py プロジェクト: sxdkxgwan/awesome_nmt
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)
コード例 #2
0
ファイル: translate.py プロジェクト: changlinzhang/nl2bash
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)