示例#1
0
 def _create_model(self, sess, basename, **kwargs):
     model = load_seq2seq_model(
         basename,
         sess=sess, predict=True,
         beam=self.task.config_params.get('beam', 30),
         **kwargs
     )
     return model, model.decoder.best, None
示例#2
0
 def _create_model(self, sess, basename, **kwargs):
     model = load_seq2seq_model(
         basename,
         sess=sess, predict=True,
         beam=self.task.config_params.get('beam', 30),
         **kwargs
     )
     return model, model.decoder.best, None
示例#3
0
def load_model(modelname, **kwargs):
    return load_seq2seq_model(BASELINE_SEQ2SEQ_LOADERS, modelname, **kwargs)