数据是N1121145121的numpy数组,N是样本数量,设置的batch为4
运行的时候出现了这个错误RuntimeError: Expected 5-dimensional input for 5-dimensional weight [64, 1, 7, 7, 7], but got 6-dimensional input of size [4, 2, 1, 121, 145, 121] instead
用torch.squeeze也没啥变化
不知道是哪个环节出现问题,期待解答,谢谢!
emm. 很明显的问题 维度不一致
要么改网络 要么改输入