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

Public Member Functions

def setUp (self)
def test_step_lr (self)
def test_multi_step_lr (self)
def test_exp_lr (self)
def test_cos_anneal_lr (self)
def test_legacy_step_lr (self)
def test_legacy_multi_step_lr (self)
def test_legacy_exp_lr (self)
def test_legacy_cos_anneal_lr (self)
def test_reduce_lr_on_plateau1 (self)
def test_reduce_lr_on_plateau2 (self)
def test_reduce_lr_on_plateau3 (self)
def test_reduce_lr_on_plateau4 (self)
def test_reduce_lr_on_plateau5 (self)
def test_reduce_lr_on_plateau6 (self)
def test_reduce_lr_on_plateau7 (self)
def test_reduce_lr_on_plateau8 (self)
def test_compound_step_and_multistep_lr (self)
def test_compound_step_and_exp_lr (self)
def test_compound_exp_and_multistep_lr (self)
def test_compound_cosanneal_and_step_lr (self)
def test_compound_cosanneal_and_multistep_lr (self)
def test_compound_cosanneal_and_exp_lr (self)
def test_compound_reduce_lr_on_plateau1 (self)
def test_compound_reduce_lr_on_plateau2 (self)
def test_compound_reduce_lr_on_plateau3 (self)
def test_compound_reduce_lr_on_plateau4 (self)
def test_lambda_lr (self)
def test_step_lr_state_dict (self)
def test_multi_step_lr_state_dict (self)
def test_exp_step_lr_state_dict (self)
def test_cosine_lr_state_dict (self)
def test_reduce_lr_on_plateau_state_dict (self)
def test_lambda_lr_state_dict_fn (self)
def test_lambda_lr_state_dict_obj (self)
- Public Member Functions inherited from common_utils.TestCase
def __init__ (self, method_name='runTest')
def assertLeaksNoCudaTensors (self, name=None)
def wrap_with_cuda_memory_check (self, method)
def setUp (self)
def assertTensorsSlowEqual (self, x, y, prec=None, message='')
def genSparseTensor (self, size, sparse_dim, nnz, is_uncoalesced, device='cpu')
def safeToDense (self, t)
def safeCoalesce (self, t)
def assertEqual (self, x, y, prec=None, message='', allow_inf=False)
def assertAlmostEqual (self, x, y, places=None, msg=None, delta=None, allow_inf=None)
def assertNotEqual (self, x, y, prec=None, message='')
def assertObjectIn (self, obj, iterable)
def assertExpectedRaises (self, exc_type, callable, args, kwargs)
def assertWarns (self, callable, msg='')
def assertWarnsRegex (self, callable, regex, msg='')
def assertExpected (self, s, subname=None)
- Public Member Functions inherited from expecttest.TestCase
def assertExpectedInline (self, actual, expect, skip=0)

Public Attributes


Additional Inherited Members

- Static Public Attributes inherited from common_utils.TestCase
int precision = 1
 maxDiff = None
 assertRegex = unittest.TestCase.assertRegexpMatches
 assertRaisesRegex = unittest.TestCase.assertRaisesRegexp
- Static Public Attributes inherited from expecttest.TestCase
bool longMessage = True

Detailed Description

Definition at line 510 of file

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