RuntimeError: Given groups=1, weight of size [512, 80, 7], expected input[1, 329, 80] to have 80 ch

完整报错:

RuntimeError: Given groups=1, weight of size [512, 80, 7], expected input[1, 329, 80] to have 80 channels, but got 329 channels instead


解析错误:

Given groups=1, weight of size [512, 80, 7]

无分组卷积,卷积核的输入通道是80,输入通道是512,大小为7(一维卷积)(如果是二维卷积,则此处的size是会有4个数,后两个表示卷积核的大小)

expected input[1, 329, 80] to have 80 channels, but got 329 channels instead

实际输入的张量维度是[1,329,80], 即feature的channel 是329,但期望的输入feature的channel为80通道

我这边的实际错误是读入input的shape是[329,80],所以仅通过

input=input.T

即转置后便解决错误,但如果直接就是不匹配的,要么选择修改网络层的结果(即输入通道设置),要么使用reshape将输入进行变化。

你可能感兴趣的:(pytorch,cnn,深度学习,人工智能)