Beispiel #1
0
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,))
Beispiel #2
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