Пример #1
0
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
Пример #2
0
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