def build_translator(opt): dummy_parser = configargparse.ArgumentParser(description='translate.py') opts.model_opts(dummy_parser) dummy_opt = dummy_parser.parse_known_args([])[0] fields, model = nmt_model.load_test_model(opt, dummy_opt.__dict__) translator = Translator(model, fields, opt) return translator
def build_translator(opt,model_path): dummy_parser = configargparse.ArgumentParser(description='translate.py') opts.model_opts(dummy_parser) dummy_opt = dummy_parser.parse_known_args([])[0] #build the model and get the checkpoint and field path=[] path.append(model_path) opt.models=path fields, model = nmt_model.load_test_model(opt, dummy_opt.__dict__) translator = Translator(model, fields, opt) return translator