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):
models, tgt_dict, beam_size=args.beam, stop_early=(not args.no_early_stop),
normalize_scores=(not args.unnormalized), len_penalty=args.lenpen,
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:
......
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