Changing batch size

Hi Nick

Thanks for sharing this piece of code.

I am trying to use the model with slightly different parameters what I noticed is as soon as I change the batch_size the code gives me an error. I am trying to find out why (despite the fact that you have parameterised the input shape while including batch size). Would you have any clues on what is going on?
Thanks
