db19408b9dba8b85a16d9a7b58c5bb366d600c7c,examples/transformer/transformer_overall.py,,,#,16

Before Change

    if args.load_from_pytorch:
        encoder_input = ori_src_text
    else:
        encoder_input = ori_src_text[:, 1:]

    decoder_input = ori_tgt_text[:, :-1]
    labels = ori_tgt_text[:, 1:]

After Change

            elif args.model_dir == "max":
                args.model_dir = args.log_dir + "./max/"
                eval_saver.restore(sess. tf.train.latest_checkpoint(args.model_dir))
                mname = tf.train.latest_checkpoint(args.model_dir).split("/")[-1]
            else:
                print("load model from {}".format(args.model_fullpath))
                mname = args.model_fullpaths.split("/")[-1]

Instances