training error while setting the batch size more than 1

[arminbiglari]

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?

[Xinyu-Fan]

I also want to ask the same question！