用改进后模型的权重进行detect.py时报错,有偿求如何解决这个问题
把你输入conv2d函数的数据用这段代码处理一下
data = data.detach().cpu().numpy()
data = torch.FloatTensor(data)
这个data就是你的数据
FloatTensor如果不行的话改成:
data=torch.HalfTensor(data)
报错:可疑的运行行为警告,没有看到你的代码,只能给你推荐类似很接近的案例给你,供你参考,链接:https://blog.csdn.net/weixin_42042056/article/details/105995301
期望能对你有所帮助
翻译:
RuntimeError:输入类型(torch.cuda.FloatTensor)和权重类型(torch.cuda.HalfTensor)应该是相同的
问题:
可能是因为您的模型在 GPU 上,但您的数据在 CPU 上。
解决方案:
检查一下模型和数据是否都在GPU上,如果不在,改为一致的即可。