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
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
def load_model(modelname, **kwargs): return load_seq2seq_model(BASELINE_SEQ2SEQ_LOADERS, modelname, **kwargs)