問題描述

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

解決方法

對net或者新添加的模塊進行轉換。

解決方法1

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

解決方法2

net = net.cuda()

總結

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

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