__init__(self, beam_size, model, eos_token_id, go_token_id=seq2seq_util.GO_ID, post_eos_penalty=None) (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |
apply(self, inputs, length, log_probs, attentions, state_configs, data_dependencies, word_rewards=None, possible_translation_tokens=None, go_token_id=None) (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |
beam_size (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |
eos_token_id (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |
get_previous_tokens(self) (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |
get_step_model(self) (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |
get_timestep(self) (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |
go_token_id (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |
LinkConfig (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | static |
model (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |
post_eos_penalty (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |
StateConfig (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | static |
step_model (defined in caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly) | caffe2.python.models.seq2seq.beam_search.BeamSearchForwardOnly | |