1. 程式人生 > 程式設計 >淺談pytorch池化maxpool2D注意事項

淺談pytorch池化maxpool2D注意事項

注意: 在搭建網路的時候用carpool2D的時候,讓高度和寬度方向不同池化時,

用如下:

nn.MaxPool2d(kernel_size=2,stride=(2,1),padding=(0,1))

千萬不要用:

nn.MaxPool2d(kernel_size=2,0)), 這樣在用交叉熵做損失函式的時候,有時候會出現loss為nan的情況,檢查的時候發現,某些樣本的提取出來的feature全為nan。

以上這篇淺談pytorch池化maxpool2D注意事項就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。