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

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

parent ac5fddfc

interactive.py

+1 −1
Changes for interactive.py: 1 added line, 1 removed line.
Original line number Diff line number Diff line
@@ -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: