pytorch RuntimeError: expected scalar type Double but found Float

论坛 期权论坛 期权     
选择匿名的用户   2021-5-29 06:35   5489   0
RuntimeError: expected scalar type Double but found Float

原因: tensor的数据类型dtype不正确

解决: 将数据类型转为float32

tensor = tensor.to(torch.float32)

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP