ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256,1,1]
解决办法:
在DataLoader类里面加上:drop_last=True

已解决。
在进行深度学习训练时,遇到了ValueError,问题在于每个通道的输入数据只有一个元素。通过在DataLoader设置`drop_last=True`,成功解决了训练过程中因输入尺寸不匹配导致的错误。
ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256,1,1]
解决办法:
在DataLoader类里面加上:drop_last=True

已解决。
888
1980

被折叠的 条评论
为什么被折叠?