RuntimeError: expected scalar type Float but found Half
把self.model = self.model.eval().to(device=‘cuda’).half()
中的.half()去掉
RuntimeError: expected scalar type Float but found Half
最新推荐文章于 2024-01-25 13:42:07 发布
文章讨论了一个在使用CUDA设备时遇到的运行时错误,该错误涉及到模型的浮点数类型不匹配。通过移除代码中将模型转换为半精度浮点(.half())的调用来解决问题。这可能导致性能影响,但可以确保数据类型的一致性。
摘要由CSDN通过智能技术生成