Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
model_defs.word_language_model.RNNModel Class Reference
Inheritance diagram for model_defs.word_language_model.RNNModel:

Public Member Functions

def __init__ (self, rnn_type, ntoken, ninp, nhid, nlayers, dropout=0.5, tie_weights=False, batchsize=2)
 
def init_weights (self)
 
def forward (self, input, hidden)
 
def init_hidden (self, bsz)
 

Static Public Member Functions

def repackage_hidden (h)
 

Public Attributes

 drop
 
 encoder
 
 rnn
 
 decoder
 
 rnn_type
 
 nhid
 
 nlayers
 
 hidden
 

Detailed Description

Container module with an encoder, a recurrent module, and a decoder.

Definition at line 9 of file word_language_model.py.

Member Function Documentation

def model_defs.word_language_model.RNNModel.repackage_hidden (   h)
static
Detach hidden states from their history.

Definition at line 47 of file word_language_model.py.


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