Ejemplo n.º 1
0
def build_predictor(args, tokenizer, symbols, model, logger=None):
    scorer = GNMTGlobalScorer(args.alpha, length_penalty='wu')

    translator = Translator(args,
                            model,
                            tokenizer,
                            symbols,
                            global_scorer=scorer,
                            logger=logger)
    return translator