Exemple #1
0
def testSoftmax():
    conf = SoftmaxConfig()
    conf.EPOCH = 20
    conf.SAVE_PER_EPOCH = 5
    cifar10 = CIFAR10('data/')
    manager = Manager(conf, cifar10, softmax)

    manager.compile()
    manager.training()
Exemple #2
0
def testGoogLeNet():
    conf = GoogLeNetConfig()
    conf.EPOCH = 100
    conf.SAVE_PER_EPOCH = 10
    conf.LEARNING_RATE = 1e-2
    cifar10 = CIFAR10('data/')
    manager = Manager(conf, cifar10, googlenet)

    manager.compile()
    manager.training()