1.在pytorch中，創建好一個模型之后，需要將模型放入cuda才能調用GPU來進行訓練。

model.to(device)

?

2.在?pytorch中，使用nn.Conv2d()時，?里面的padding的值是通過自己設置的。

這里的dilation是膨脹卷積，一般默認值是1.通過這個公式就可以反推出P的值了。

總結

以上是生活随笔為你收集整理的pytorch注意事项的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。