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

Public Member Functions

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)
def save (self, checkpoint_path_prefix, current_step)

Public Attributes


Detailed Description

Definition at line 114 of file

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