Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
caffe2.python.models.seq2seq.translate.Seq2SeqModelCaffe2EnsembleDecoder Class Reference
Inheritance diagram for caffe2.python.models.seq2seq.translate.Seq2SeqModelCaffe2EnsembleDecoder:

Public Member Functions

def scope (self, scope_name, blob_name)
 
def build_word_rewards (self, vocab_size, word_reward, unk_reward)
 
def __init__ (self, translate_params)
 
def load_models (self)
 
def decode (self, numberized_input, max_output_seq_len)
 

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 49 of file translate.py.


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