Train halfway RuntimeError: The size of tensor a (4) must match the size of tensor b (2) at non-singleton dimension 0
xiaohongzaizhe opened this issue · 0 comments
xiaohongzaizhe commented
Train halfway RuntimeError: The size of tensor a (4) must match the size of tensor b (2) at non-singleton dimension 0