Expected more than 1 value per channel when training, got input size torch.Size([1, 100])

Question

Expected more than 1 value per channel when training, got input size torch.Size([1, 100])

Closed this issue 3 years ago · 1 comments

wb1997-cell commented 3 years ago

Expected more than 1 value per channel when training, got input size torch.Size([1, 100])
I just changed the batch from 128 to 8, because MY GPU is not enough, but there is a mistake above, I don't know why

Answer 1 · 2022-01-04T05:58:43.000Z

Plz post the err details