search

FlorianMuellerklein / lasagne_mnist

Convolution Neural Network written in Theano/Lasagne for MNIST
7 stars 2 forks source link

Theano and Lasagne Version #1

Open trebuchet90 opened 7 years ago

trebuchet90 commented 7 years ago

Hi Florian, What versions of Theano and Lasagne did this work with?

FlorianMuellerklein commented 7 years ago

To get it to work with new versions I think that you just need to change:

def lasagne_model():
    l_in = InputLayer(shape=(None, 1, 28, 28))

to:

def lasagne_model(input_var=None):
    l_in = InputLayer(shape=(None, 1, 28, 28), input_var=input_var)