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

// 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.")
beam_search_parser.add_argument("--k", type=int, default=5, help="K in the length penalty term.")

// random sampler options
random_sample_parser = subparsers.add_parser("random-sample",
                                             help="Use random sampling for decoding.")
random_sample_parser.add_argument("--temperature", type=float, default=1.0,
                                  help="Softmax temperature used in sampling.")
random_sample_parser.add_argument("--use-top-k", type=int, required=False,
                                  help="Sample only from the top-k candidates.")

Instances