Commit eaa576b0 authored by Sergey Edunov's avatar Sergey Edunov
Browse files

Pass sampling-temperature trough to the generator in interactive.py

parent ac5fddfc
...@@ -90,7 +90,7 @@ def main(args): ...@@ -90,7 +90,7 @@ def main(args):
models, tgt_dict, beam_size=args.beam, stop_early=(not args.no_early_stop), models, tgt_dict, beam_size=args.beam, stop_early=(not args.no_early_stop),
normalize_scores=(not args.unnormalized), len_penalty=args.lenpen, normalize_scores=(not args.unnormalized), len_penalty=args.lenpen,
unk_penalty=args.unkpen, sampling=args.sampling, sampling_topk=args.sampling_topk, unk_penalty=args.unkpen, sampling=args.sampling, sampling_topk=args.sampling_topk,
minlen=args.min_len, minlen=args.min_len, sampling_temperature=args.sampling_temperature
) )
if use_cuda: if use_cuda:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment