Hi, Marc,
I got the same error during loss.backward().
RuntimeError: The size of tensor a (501) must match the size of tensor b (434) at non-singleton dimension 0
Have you solved this error? Any hint?
Thanks