Expected object of device type cuda but got device type cpu for argument
2020/03/31 下午 09:06
電腦視覺深度學習討論版
Sherry
觀看數：6
回答數：1
收藏數：0
cvdl-1
cvdl-1-d29
在執行程式碼，不知道發生什麼問題，想請教怎麼處理。
回答列表
-
2020/03/31 下午 10:42楊哲寧贊同數：1不贊同數：0留言數：0
您好，這裡必需先將x轉成cuda ，請多加一行
x.type(torch.cuda.FloatTensor)