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

Public Member Functions

def __init__ (self)
 
def forward (self, x)
 

Public Attributes

 conv1
 
 conv2
 
 conv2_drop
 
 fc1
 
 fc2
 

Detailed Description

Definition at line 5 of file mnist.py.


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