Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | List of all members
test_nn.NewCriterionTest Class Reference
Inheritance diagram for test_nn.NewCriterionTest:
test_nn.InputVariableMixin common_nn.CriterionTest common_nn.TestBase

Public Member Functions

def __init__ (self, args, kwargs)
 
def test_cuda (self, test_case, dtype=None, extra_args=None)
 
def constructor_args (self)
 
def extra_args (self)
 
- Public Member Functions inherited from common_nn.CriterionTest
def __init__ (self, args, kwargs)
 
def __call__ (self, test_case)
 
def test_cuda (self, test_case)
 
- Public Member Functions inherited from common_nn.TestBase
def __init__ (self, constructor, desc='', reference_fn=None, fullname=None, kwargs)
 
def get_name (self)
 
def constructor_args (self)
 
def extra_args (self)
 
def __call__ (self, test_case)
 

Public Attributes

 check_gradgrad
 
 check_half
 
 convert_target
 
- Public Attributes inherited from common_nn.CriterionTest
 should_test_cuda
 
 check_forward_only
 
- Public Attributes inherited from common_nn.TestBase
 desc
 
 fullname
 
 constructor
 
 reference_fn
 

Detailed Description

Definition at line 404 of file test_nn.py.


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