pytorch踩坑記


因為我有數學物理背景,所以清楚卷積的原理。但是在看pytorch文檔的時候感到非常頭大,羅列的公式以及各種令人眩暈的下標讓入門新手不知所雲...最初我以為torch.nn.conv1d的參數in_channel/out_channel表示圖像的通道數,經過運行錯誤提示之后,才知道[in_channel,kernel_size]構成了卷積核。 

loss函數中target/label的類型必須是dtype=torch.long 

在使用cross_entropy時出現了一個錯誤RuntimeError: Dimension out of range (expected to be in range of [-1, 0], but got 1),查看functional.py的1550行中:
其中log_softmax(input, 1)指定了dim=1,而log_softmax實現如下:

排查之后發現,cross_entropy輸入的矩陣行數需要等於taget/label的個數,即行對行。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



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