def unittest():
for dtype, atol in dtypesSupported():
calcTest(dtype, atol)
trainSimpleTest(RMSPropGraves, dtype, learnRate=1e-2)
if Config.backend == Config.Backend.cuda:
trainHardTest(RMSPropGraves, dtype, learnRate=1e-2)
def unittest():
for dtype, atol in dtypesSupported():
calcTest(dtype, atol)
trainSimpleTest(NesterovSGD, dtype, learnRate=1e-1, momRate=0.9)
if Config.backend == Config.Backend.cuda:
trainHardTest(NesterovSGD, dtype, learnRate=1e-1, momRate=0.9)
def unittest():
for dtype, atol in dtypesSupported():
calcTest(dtype, atol)
trainSimpleTest(Adam, dtype, alpha=1e-2)
if Config.backend == Config.Backend.cuda:
trainHardTest(Adam, dtype, alpha=1e-2)