ERROR if i change batch size bigger than 2

[ChrisHRZ]

if i change batch size bigger than 2, there is a error like this:
RuntimeError: The size of tensor a (10) must match the size of tensor b (2) at non-singleton dimension 1
how can i fix it?thank you!

[huangkuns]

In our experiment, the batch_size is always 1, so in our code, we do not support batch_size > 1. If you need to increase the batch_size, you need to change the several lines of code related to dataloader and loss functions.