RuntimeError: mat1 dim 1 must match mat2 dim 0
這個錯誤的大概意思是:矩陣mat1 的第二維度要與mat2的第一維度不匹配
在新增別的數(shù)據(jù)集進(jìn)行訓(xùn)練時報當(dāng)前錯誤,原因是輸入的圖像大小與之前不一樣,這是新手在學(xué)習(xí)時常會遇到的問題。文章來源:http://www.zghlxwxcb.cn/news/detail-612187.html
解決方法:
先看報錯信息,確定報錯位置
我的這個代碼是utils.py
文件的104
行 x=self.fc1(x)
這個代碼出現(xiàn)了問題,查找之后發(fā)現(xiàn)fc1
是個全連接層
咱們在104報錯的位置打上斷點,用debug進(jìn)行調(diào)試
下面是調(diào)試后顯示的變量情況
展開x
,將shape里的2304
填入之前的fc1
全連接層中,
問題解決文章來源地址http://www.zghlxwxcb.cn/news/detail-612187.html
到了這里,關(guān)于RuntimeError: mat1 dim 1 must match mat2 dim 0 解決方法的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!