錯(cuò)誤信息
類型錯(cuò)誤, 計(jì)算loss值的函數(shù)傳入的參數(shù)類型不統(tǒng)一。
解決方法
查看上文loss計(jì)算代碼部分的參數(shù)類型,如loss=f.mse_loss(out,label),檢查out和label的類型都是torch.float類型即可。使用label.dtype查看tensor的類型。
具體流程
報(bào)錯(cuò)定位在這一行
尋思著是否是loss類型的問題,于是我就添加文章來源:http://www.zghlxwxcb.cn/news/detail-579105.html
loss = loss.to(torch.float32)
但是還是報(bào)錯(cuò)在此處,于是開始考慮是否是上文loss計(jì)算代碼部分的參數(shù)類型有問題
添加這兩行,再次運(yùn)行,sucess, 完美文章來源地址http://www.zghlxwxcb.cn/news/detail-579105.html
到了這里,關(guān)于loss.backward()處遇到“RuntimeError: Found dtype Double but expected Float”的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!