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

// 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.")

// shared options
for p in [beam_search_parser, random_sample_parser]:
    p.add_argument("--gpu", type=int, default=0,

Instances