ValueError: could not broadcast input array from shape (256,256,3) into shape (512,512,3)

[codegeniur]

Hi pannous, like your effort in putting caffe speech recognition examples here!

I managed to run the training and got the model. I run python/classify.py to get the output but got the following error:
ValueError: could not broadcast input array from shape (256,256,3) into shape (512,512,3)
(have tried with the ipynb and the broadcasting error occurs too)

Any idea if this is an issue with the training parameters (e.g. input images have to be 256x256) or is this an issue with how I run the classify.py?

Many thanks in advance!