Hi Victor,

I am using a network which produces 32x32 images due to the mathematical simplicity of such an approach. The MNIST dataset is being padded from 28x28 to 32x32. Since the edges are black for all MNIST digits, this doesn’t change the dataset in any meaningful way.

Can you tell me exactly which line of code is causing this error? The code is working fine on the two computers I run it on.

PhD. Interests include Deep (Reinforcement) Learning, Computational Neuroscience, and Phenomenology.

Get the Medium app

A button that says 'Download on the App Store', and if clicked it will lead you to the iOS App store
A button that says 'Get it on, Google Play', and if clicked it will lead you to the Google Play store