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

Public Member Functions

def exportTest (self, model, inputs, rtol=1e-2, atol=1e-7)
 
def test_ops (self)
 
def test_prelu (self)
 
def test_concat (self)
 
def test_permute (self)
 
def test_srresnet (self)
 
def test_super_resolution (self)
 
def test_alexnet (self)
 
def test_mnist (self)
 
def test_vgg16 (self)
 
def test_vgg16_bn (self)
 
def test_vgg19 (self)
 
def test_vgg19_bn (self)
 
def test_resnet (self)
 
def test_inception (self)
 
def test_squeezenet (self)
 
def test_densenet (self)
 
def test_dcgan_netD (self)
 
def test_dcgan_netG (self)
 

Detailed Description

Definition at line 47 of file test_models.py.


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