第Batch[100]次时候报错

[githupdebaba]

报错：
RuntimeError: Expected 4-dimensional input for 4-dimensional weight [100, 1, 3, 128], but got 3-dimensional input of size [2495, 1, 128] instead

输出size
Batch[9] - loss: 0.666672 acc: 86.7188%(111/128)1 torch.Size([128, 1, 103, 128])
torch.Size([128, 6])
Batch[10] - loss: 0.486642 acc: 90.6250%(116/128)1 torch.Size([2495, 1, 128])

[woshicqy]

老哥，你改了其他地方吗？我一开始的epoch，他的code就有shape上的问题。。。。