RuntimeError: non-positive stride is not supported
在使用pytorch中的nn.Conv3d时遇到 RuntimeError: non-positive stride is not supported
代码如下:
nn.Conv3d(in_channels, out_channels, kernel_size=1, padding=0)
问题原因:
此时padding为0不符合参数要求,torch会把padding=0当成是stride的参数,而stride不能为0,因此报错!!!
只需要修改为:
```python
nn.Conv3d(in_channels, out_channels, kernel_size=1)
即可。
卷积核为3时对应的代码为:
nn.Conv3d(in_channels, out_channels, kernel_size=3, padding=1)