Commit 4ce453b1 authored by Sergey Edunov's avatar Sergey Edunov Committed by Myle Ott
Browse files

Sampling doesn't work with interactive

parent a04c4cf4
Loading
Loading
Loading
Loading
+1 −1
Changes for interactive.py: 1 added line, 1 removed line.
Original line number Diff line number Diff line
@@ -41,7 +41,7 @@ def main(args):
    translator = SequenceGenerator(
        models, beam_size=args.beam, stop_early=(not args.no_early_stop),
        normalize_scores=(not args.unnormalized), len_penalty=args.lenpen,
        unk_penalty=args.unkpen)
        unk_penalty=args.unkpen, sampling=args.sampling)
    if use_cuda:
        translator.cuda()