RuntimeError: size mismatch, m1: [80 x 4], m2: [320 x 50] at …\aten\src\TH/generic/THTensorMath.cpp:41

使用pytorch進(jìn)行深度學(xué)習(xí)的訓(xùn)練會(huì)出現(xiàn)這種問(wèn)題，原因是fc全連接層的輸入維度問(wèn)題，由于輸入是二維的數(shù)據(jù)，很多時(shí)候在輸入全連接層的時(shí)候我們沒(méi)有調(diào)整維度
例如：

def forward(self, input):x = self.pool1(F.relu(self.conv1(input)))x = self.pool2(F.relu(self.conv2(x)))x = self.fc1(x)x = self.fc2(x)return x

我們的fc1的輸入將是804的情況，而需要的是3201的情況，這時(shí)會(huì)報(bào)錯(cuò)

改為：

def forward(self, input):x = self.pool1(F.relu(self.conv1(input)))x = self.pool2(F.relu(self.conv2(x))).view(320)x = self.fc1(x)x = self.fc2(x)return x

使用view調(diào)整維度即可

總結(jié)

以上是生活随笔為你收集整理的RuntimeError: size mismatch, m1: [80 x 4], m2: [320 x 50] at ..\aten\src\TH/generic/THTensorMath.cpp的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。