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

Public Member Functions

def __init__ (self, args)
 
def CreateModel (self)
 
def TrainModel (self)
 
def GenerateText (self, num_characters, ch)
 

Public Attributes

 seq_length
 
 batch_size
 
 iters_to_report
 
 hidden_size
 
 text
 
 vocab
 
 char_to_idx
 
 idx_to_char
 
 D
 
 cell_state
 
 forward_net
 
 model
 
 predictions
 
 loss
 
 prepare_state
 

Detailed Description

Definition at line 53 of file char_rnn.py.


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