I tried to use a batch size of more than 1 to train TransVOD Lite using a custom dataset. But I got the following error RuntimeError: The size of tensor a (28) must match the size of tensor b (14) at non-singleton dimension 0 . Is there a solution to this problem? Or maybe there is another solution so that I can speed up the training process a little.
I tried to use a batch size of more than 1 to train TransVOD Lite using a custom dataset. But I got the following error
RuntimeError: The size of tensor a (28) must match the size of tensor b (14) at non-singleton dimension 0
. Is there a solution to this problem? Or maybe there is another solution so that I can speed up the training process a little.