pytorch注意事项

1.在pytorch中，创建好一个模型之后，需要将模型放入cuda才能调用GPU来进行训练。

model.to(device)

 

2.在 pytorch中，使用nn.Conv2d()时， 里面的padding的值是通过自己设置的。

这里的dilation是膨胀卷积，一般默认值是1.通过这个公式就可以反推出P的值了。