python - pytorch cnn 步幅错误
问题描述
我现在在 Windows 中使用 pytorch 0.4.0 来构建 CNN,这是我的代码:
class net(nn.Module):
def __init__(self):
super(net, self).__init__()
self.conv1 = nn.Conv2d(in_channels=1, out_channels=16, kernel_size=(1,3),stride=1 )
self.conv2 = nn.Conv2d(in_channels=16, out_channels=32, kernel_size=(1,3), stride=1)
self.dense1 = nn.Linear(32 * 28 * 24, 60)
self.out = nn.Linear(60,3)
def forward(self, input):
x = F.relu(self.conv1(input))
x = F.relu(self.conv2(x))
x = x.view(x.size(0), -1) # flatten(batch,32*7*7)
x = self.dense1(x)
output = self.out(x)
return output
但我得到的错误是
File "D:\Anaconda\lib\site-packages\torch\nn\modules\conv.py", line 301, in forward
self.padding, self.dilation, self.groups)
RuntimeError: expected stride to be a single integer value or a list of 1 values to match the convolution dimensions, but got stride=[1, 1]
我认为这表明我在上面的代码中犯了一些错误,但我不知道如何修复它,任何人都可以帮助我吗?提前致谢!
解决方案
推荐阅读
- sql-server - Visual Studio SSRS 数据源:Windows 身份验证 - 使用此用户名和密码
- python - 多次重置/重新安装后无法将 numpy 导入 python
- javascript - Javascript - 使用 Extend(underscore.js 库)在对象中通过引用传递
- java - Spring Security 应用程序的用户类
- intellij-idea - 如何使用 ThreadDumpVisualizer IntelliJ IDEA 插件分析 Java 线程转储
- git - 如何中途进入git项目并了解树形结构
- docker - Docker:为什么我需要在 Ubuntu 中使用 sudo?
- python - Matplotlib 子图
- angular - Angular 可观察数组,由 http 请求和手动添加的数据填充
- windows - 如何阻止 WinDbg 缩小其窗口?