通过查看代码找到了问题所在是因为使用了不正确的BatchNorm函数
原因分析
我们可以看到官方文档对于BatchNorm1d的解释
BatchNorm1d
显然需要输入的是二维或者三维,但是我们有一个四维的输入
那么我们可以查看BatchNorm2d的官方解释
BatchNorm2d
可以接受四维输入,
解决方法
根据上述分析,把BatchNorm1d改为BatchNorm2d即可
通过查看代码找到了问题所在是因为使用了不正确的BatchNorm函数
我们可以看到官方文档对于BatchNorm1d的解释
BatchNorm1d
显然需要输入的是二维或者三维,但是我们有一个四维的输入
那么我们可以查看BatchNorm2d的官方解释
BatchNorm2d
可以接受四维输入,
根据上述分析,把BatchNorm1d改为BatchNorm2d即可