Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
seq2seq.Seq2SeqModelCaffe2 Class Reference

Public Member Functions

def output_projection (self, model, decoder_outputs, decoder_output_size, target_vocab_size, decoder_softmax_size)
 
def model_build_fun (self, model, forward_only=False, loss_scale=None)
 
def forward_model_build_fun (self, model, loss_scale=None)
 
def norm_clipped_grad_update (self, model, scope)
 
def norm_clipped_sparse_grad_update (self, model, scope)
 
def total_loss_scalar (self)
 
def __init__ (self, model_params, source_vocab_size, target_vocab_size, num_gpus=1, num_cpus=1)
 
def __enter__ (self)
 
def __exit__ (self, exc_type, exc_value, traceback)
 
def initialize_from_scratch (self)
 
def get_current_step (self)
 
def inc_current_step (self)
 
def step (self, batch, forward_only)
 

Public Attributes

 num_gpus
 
 model
 
 forward_net
 
 encoder_type
 
 learning_rate
 
 global_step
 
 start_time
 
 encoder_embeddings
 
 decoder_embeddings
 
 model_params
 
 encoder_params
 
 source_vocab_size
 
 target_vocab_size
 
 num_cpus
 
 batch_size
 

Detailed Description

Definition at line 106 of file seq2seq.py.


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