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

當模型中有nn.BatchNorm時，batch不能設置為1，也不能出現樣本最后一個batch可能就剩一個樣本

解決方法

• batch_size > 1
• DataLoader中設置 drop_last=True

總結

以上是生活随笔為你收集整理的Debug Pytorch: ValueError: Expected more than 1 value per channel when training, got input size tor的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。