1.在pytorch中，創(chuàng)建好一個模型之后，需要將模型放入cuda才能調(diào)用GPU來進行訓(xùn)練。

model.to(device)

?

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

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。