原文:RuntimeError: expected backend CPU and dtype Double but got backend CPU and dtype Float

此問題是由於數據類型不匹配導致的,在出錯的數據后面添加.double ...

2020-11-03 12:21 0 441 推薦指數:

查看詳情

【E-14-Pytorch】RuntimeError: Expected object of device type cuda but got device type cpu for argument #2 'mat1' in call to _th_addmm

一、出錯誤背景: Pytorch 中想使用 CUDA 對程序計算進行加速 二、問題分析 錯誤的意思:object 的 device 類型期望得到的是 cuda 類型,但是實際上的類型確實 cpu 類型,在調用二分類交叉熵損失進行前向計算的時候 三、檢查下面幾點: 模型是否放到 ...

Sat Oct 24 02:37:00 CST 2020 0 4751
 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM