cifar10.py 文件源码

def build_network(layers):
    """
    Builds and returns a NeuralNet object
    """
    return  NeuralNet(
        layers = layers,

        update = nesterov_momentum,
        update_learning_rate = theano.shared(float32(0.01)),
        update_momentum = theano.shared(float32(0.9)),    

        on_epoch_finished=[
            AdjustVariable('update_learning_rate', start=0.01, stop=0.0001),
            AdjustVariable('update_momentum', start=0.9, stop=0.999),
            EarlyStopping(patience=200),
            ],

        # Automatically split the data (90% training 10% validation).
        train_split = TrainSplit(eval_size=0.10), 

        regression = False,
        max_epochs = 300,
        verbose = 2, 
        )

# --------------- Training the network ---------------