def testSoftmax(): conf = SoftmaxConfig() conf.EPOCH = 20 conf.SAVE_PER_EPOCH = 5 cifar10 = CIFAR10('data/') manager = Manager(conf, cifar10, softmax) manager.compile() manager.training()
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()