Sampling doesn't work with interactive

4ce453b1 · Sergey Edunov · Myle Ott · a04c4cf4 · 4ce453b1
Commit 4ce453b1 authored May 10, 2018 by Sergey Edunov Committed by Myle Ott Jun 15, 2018
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

interactive.py interactive.py +1 -1

No files found.
--- a/interactive.py
+++ b/interactive.py
@@ -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()