training error while setting the batch size more than 1

[fffffq99]

hello, thanks for your great work. i am trying to run your code on my own dataset, but there is an error. when i set batch size to a number greater than 1, the training process crashes, whereas, when i set it to 1 it goes fine.

do you know what is the problem?