caffe2.python.models.seq2seq.translate.Seq2SeqModelCaffe2EnsembleDecoder Class Reference

Inheritance diagram for caffe2.python.models.seq2seq.translate.Seq2SeqModelCaffe2EnsembleDecoder:

Public Member Functions
def	get_model_file (self, model)
def	get_db_type (self)
def	scope (self, scope_name, blob_name)
def	__init__ (self, translate_params)
def	decode (self, numberized_input, max_output_seq_len)
Public Member Functions inherited from caffe2.python.models.seq2seq.translate.Seq2SeqModelCaffe2EnsembleDecoderBase
def	get_model_file (self, model)
def	get_db_type (self)
def	build_word_rewards (self, vocab_size, word_reward, unk_reward)
def	load_models (self)

Public Attributes
	models
	beam_size
	source_vocab_size
	target_vocab_size
	decoder_scope_names
	model
	encoder_inputs
	encoder_lengths
	max_output_seq_len

Detailed Description

Definition at line 83 of file translate.py.

---

The documentation for this class was generated from the following file:

• caffe2/python/models/seq2seq/translate.py