ERROR if i change batch size bigger than 2
ChrisHRZ opened this issue · comments
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!
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.