原文:RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 1 #805

具體錯誤日志如下: 解決辦法: 你輸入的圖像數據的維度不完全是一樣的,比如是訓練的數據有 組,其中 組是 ,但有一組是 ,這樣會導致Pytorch的檢查程序報錯 另外一個則是比較隱晦的batchsize的問題,Pytorch中檢查你訓練維度正確是按照每個batchsize的維度來檢查的,比如你有 組數據 假設每組數據為三通道 px px的圖像 ,batchsize為 ,那么每次訓練則提取 , , ...

2020-01-25 01:38 0 1553 推薦指數:

查看詳情

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