原文:深度學習報錯:Expected more than 1 value per channel when training, got input size torch.Size([1, 256])

造成該結果的操作:修改batch size to 根本原因:模型中用了batchnomolization,訓練中用batch訓練的時候當前batch恰好只含一個sample,而由於BatchNorm操作需要多於一個數據計算平均值,因此造成該錯誤。 解決方法:在torch.utils.data.DataLoader類中或自己創建的繼承於DataLoader的類中設置參數drop last True ...

2022-03-24 16:48 0 2722 推薦指數:

查看詳情

 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM