aeb516494daa207720b428072ab49eeea7bfff75,scripts/text_generation/sequence_sampling.py,,,#,33

Before Change

parser.add_argument("--temperature", type=float, default=1.0,
                    help="Softmax temperature used in sampling.")
args = parser.parse_args()
if not args.use_beam_search and not args.use_sampling:
    parser.error("No decoding option specified. Add one of --use-sampling and --use-beam-search.")

print(args)
if args.gpu is None:
    ctx = mx.cpu()
else:

After Change

                                             "WikiText as our encoder.")

// beam search sampler options
subparsers = parser.add_subparsers(help="Sequence generation methods.",
                                   dest="command")
subparsers.required = True
beam_search_parser = subparsers.add_parser("beam-search", help="Use beam search for decoding.")
beam_search_parser.add_argument("--alpha", type=float, default=0.0,
                                help="Alpha in the length penalty term.")

Instances