1. 程式人生 > 其它 >PyTorch:expected scalar type Float but found Double

PyTorch:expected scalar type Float but found Double

技術標籤:機器學習

原因分析:

程式碼中網路引數型別不統一

解決方案:

在最前面加

import torch
torch.set_default_tensor_type(torch.DoubleTensor)

或者在網路初始化之後加=

net = net.double()

轉載:https://blog.csdn.net/sazass/article/details/109725458