天天看點

RuntimeError: Given groups=1, weight of size 64 128 1 7, expected input[16,

如果輸入和輸出比對的情況下有這樣的錯誤,請檢查在定義卷積的時候有沒有名字重複的,

例:

("2conv1,3_bn1", ConvBN(channelNum * 4, channelNum * 2, [1, 3])),

2conv1,3_bn1這個是定義卷積的名字,如果存在重複的,則可能引起這樣的錯誤。

繼續閱讀