Ask a Question
Back to All

Theano error when running Quick example

When running the Quick example of the MLP on the MNIST data I am getting the following error when running on Python 3.5:

Using gpu device 0: GeForce GT 650M (CNMeM is disabled, cuDNN not available)
Getting data...
Creating model...
Traceback (most recent call last):

File "", line 1, in
runfile('/Users/molinal/SkyDrive/AI/opendeep/mlp_opendeep.py', wdir='/Users/molinal/SkyDrive/AI/opendeep')

File "/Users/molinal/anaconda/envs/tensorflow/lib/python3.5/site-packages/spyderlib/widgets/externalshell/sitecustomize.py", line 699, in runfile
execfile(filename, namespace)

File "/Users/molinal/anaconda/envs/tensorflow/lib/python3.5/site-packages/spyderlib/widgets/externalshell/sitecustomize.py", line 88, in execfile
exec(compile(open(filename, 'rb').read(), filename, 'exec'), namespace)

File "/Users/molinal/SkyDrive/AI/opendeep/mlp_opendeep.py", line 34, in

File "/Users/molinal/OpenDeep/opendeep/optimization/optimizer.py", line 380, in train
self.STOP = self._perform_one_epoch(f_learn, plot)

File "/Users/molinal/OpenDeep/opendeep/optimization/optimizer.py", line 422, in _perform_one_epoch
_outs = raise_to_list(f_learn(*batch))

File "/Users/molinal/anaconda/pkgs/theano/compile/function_module.py", line 786, in call

File "/Users/molinal/anaconda/pkgs/theano/tensor/type.py", line 177, in filter

TypeError: ('Bad input argument to theano function with name "f_learn" at index 0(0-based)', 'Wrong number of dimensions: expected 2, got 4 with shape (100, 1, 28, 28).')