這個問題的主要原因是輸入的數據類型與網絡參數的類型不符。

Input type為torch.cuda.FloatTensor（GPU數據類型）， weight type（即net.parameters）為torch.FloatTensor（CPU數據類型）

有以下兩種方法

方法一

device = torch.device('cuda:0') net.to(device)

方法二：

net = net.cuda()

總結

以上是生活随笔為你收集整理的Pytorch出现RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。