国产 无码 综合区,色欲AV无码国产永久播放,无码天堂亚洲国产AV,国产日韩欧美女同一区二区

Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same

這篇具有很好參考價(jià)值的文章主要介紹了Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same。希望對(duì)大家有所幫助。如果存在錯(cuò)誤或未考慮完全的地方,請(qǐng)大家不吝賜教,您也可以點(diǎn)擊"舉報(bào)違法"按鈕提交疑問。

RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same or input should be a MKLDNN tensor and weight is a dense tensor


錯(cuò)誤原因:

這個(gè)錯(cuò)誤提示表明輸入數(shù)據(jù)的類型和模型權(quán)重的類型不匹配,可能是因?yàn)槟P蜋?quán)重是在GPU上訓(xùn)練的,而輸入數(shù)據(jù)是在CPU上進(jìn)行的。

可以使用model.to('cpu')將模型權(quán)重轉(zhuǎn)移到CPU上,然后再對(duì)輸入數(shù)據(jù)進(jìn)行處理。文章來源地址http://www.zghlxwxcb.cn/news/detail-739826.html

model_path = '模型路徑'
model = torch.load(model_path)
model.to('cpu')

到了這里,關(guān)于Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same的文章就介紹完了。如果您還想了解更多內(nèi)容,請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!

本文來自互聯(lián)網(wǎng)用戶投稿,該文觀點(diǎn)僅代表作者本人,不代表本站立場(chǎng)。本站僅提供信息存儲(chǔ)空間服務(wù),不擁有所有權(quán),不承擔(dān)相關(guān)法律責(zé)任。如若轉(zhuǎn)載,請(qǐng)注明出處: 如若內(nèi)容造成侵權(quán)/違法違規(guī)/事實(shí)不符,請(qǐng)點(diǎn)擊違法舉報(bào)進(jìn)行投訴反饋,一經(jīng)查實(shí),立即刪除!

領(lǐng)支付寶紅包贊助服務(wù)器費(fèi)用

相關(guān)文章

覺得文章有用就打賞一下文章作者

支付寶掃一掃打賞

博客贊助

微信掃一掃打賞

請(qǐng)作者喝杯咖啡吧~博客贊助

支付寶掃一掃領(lǐng)取紅包,優(yōu)惠每天領(lǐng)

二維碼1

領(lǐng)取紅包

二維碼2

領(lǐng)紅包