解决方法:
tensor = tensor.to(torch.float32)
RuntimeError: expected scalar type Half but found Float
最新推荐文章于 2024-01-25 13:42:07 发布
解决方法:
tensor = tensor.to(torch.float32)