当前位置:   article > 正文

expected scalar type float but found half

expected scalar type float but found half

expected scalar type float but found half

这个问题很明显就是网络内的参数类型不同意;
修改:
在前面添加:这个我测试没成功

  1. torch.set_default_tensor_type(torch.DoubleTensor)
  2. torch.set_default_tensor_type(torch.FloatTensor)
  • 1

或者,在运行网络前添加:

net = net.double()

我的解决方法:

在aimet训练的时候报错,

直接找到报错的地方,加上data.float() 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/菜鸟追梦旅行/article/detail/649840
推荐阅读
相关标签
  

闽ICP备14008679号