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.