Python trainSimpleTest Examples

Example #1

File: RMSPropGraves.py Project: rsarbaev/PuzzleLib

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)

Example #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)

Example #3

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)