您好,登錄后才能下訂單哦!
注意: 在搭建網絡的時候用carpool2D的時候,讓高度和寬度方向不同池化時,
用如下:
nn.MaxPool2d(kernel_size=2, stride=(2, 1), padding=(0, 1))
千萬不要用:
nn.MaxPool2d(kernel_size=2, stride=(2, 1), padding=(0, 0)), 這樣在用交叉熵做損失函數的時候,有時候會出現loss為nan的情況,檢查的時候發現,某些樣本的提取出來的feature全為nan。
以上這篇淺談pytorch池化maxpool2D注意事項就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。