赞
踩
expected scalar type float but found half
这个问题很明显就是网络内的参数类型不同意;
修改:
在前面添加:这个我测试没成功
- torch.set_default_tensor_type(torch.DoubleTensor)
-
- torch.set_default_tensor_type(torch.FloatTensor)
-
或者,在运行网络前添加:
net = net.double()
我的解决方法:
在aimet训练的时候报错,
直接找到报错的地方,加上data.float()
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。