用GPU訓練的模型在本地CPU無法進行加載

最近遇到的問題是，在GPU訓練的模型無法在本地的CPU下跑，開始一直以為是代碼的問題，后來查資料發(fā)現(xiàn)才知道通過GPU訓練的模型在cpu下無法跑 需要存成CPU的模型

我是在train.py中的

model=torch.load('xx.pkl')

改為

model = torch.load('xxx.pkl', map_location='cpu')

也就是在后邊加了一句話，應該是沒問題了，等會訓練完后看看有么有問題

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
平臺聲明：文章內(nèi)容（如有圖片或視頻亦包括在內(nèi)）由作者上傳并發(fā)布，文章內(nèi)容僅代表作者本人觀點，簡書系信息發(fā)布平臺，僅提供信息存儲服務。

推薦閱讀更多精彩內(nèi)容