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


Detailed Description

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

Definition at line 9 of file

Member Function Documentation

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

Definition at line 47 of file

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