Kernel size can‘t be greater than actual input size的解决方案

报错信息:RuntimeError: Calculated padded input size per channel: (49). Kernel size: (64). Kernel size can't be greater than actual input size

解决方案:不用修改卷积核大小,修改stride和padding即可

修改前:

self.conv1 = nn.Conv1d(in_channels=3, out_channels=3, kernel_size=64, stride=16, padding=24)

修改后:

self.conv1 = nn.Conv1d(in_channels=3, out_channels=3, kernel_size=64, stride=2, padding=60)

猜你喜欢

转载自blog.csdn.net/weixin_61745097/article/details/128642752
今日推荐