ValueError: total size of new array must be unchanged

[ramyabandaru]

I have made certain changes in the "convolutional_mlp.py" code so that it can train CIFAR-10 dataset(images are 3 * 32 * 32 size) and it is executing without any error but while I am trying to load the pickled model to predict the class labels I am getting some error and I am unable to figure it out. I am attaching the screenshot of error and the modified file.
Thanks.
covolutional.txt
predict.txt
logistic_sgd (1).txt