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