def run(template, path): cifar = prepare_cifar10() cifar_train = cifar.train cifar_train_stream = cifar.train_stream cifar_validation = cifar.validation cifar_validation_stream = cifar.validation_stream cifar_test = cifar.test cifar_test_stream = cifar.test_stream print("Compiling...", end=" ") sys.stdout.flush() network = compile(template) print("DONE") sys.stdout.flush() train(network, cifar_train_stream, cifar_validation_stream, 1e-3, 0.7, path) print("Test error rate is %f%%" % (compute_error_rate(cifar_test_stream, network.predict) * 100.0,))
#!/usr/bin/python3 from __future__ import print_function import theano import theano.tensor.signal.downsample from utils import * import time import pickle import sys from layers import * from prepare_cifar10 import * cifar = prepare_cifar10() cifar_train = cifar.train cifar_train_stream = cifar.train_stream cifar_validation = cifar.validation cifar_validation_stream = cifar.validation_stream cifar_test = cifar.test cifar_test_stream = cifar.test_stream def compile(template): X = theano.tensor.tensor4('X', dtype = 'float32') Y = theano.tensor.matrix('Y', dtype = 'uint8') model_parameters = template.params