解决pytorch报错RuntimeError: exp_vml_cpu not implemented for ‘Byte‘问题:

当在PyTorch中遇到RuntimeError: exp_vml_cpu未实现错误时,原因是尝试在'Byte'类型上执行运算。解决方法是检查并转换Tensor的数据类型为torch.float32。可以通过.tensor.dtype检查类型,并使用官方文档提供的方法进行类型转换。
摘要由CSDN通过智能技术生成

解决pytorch报错RuntimeError: exp_vml_cpu not implemented for 'Byte’问题:

在调试代码过程中遇到报错:

RuntimeError: exp_vml_cpu not implemented for 'Byte'

通过提示可知,报错是因为exp_vml_cpu 不能用于Byte类型计算,这里通过 .dtype 来查看要运算的tensor类型:

print(outputs.dtype)

输出:

torch.uint8

而在计算中,默认采用 torch.float32 进行计算,因此需要将tensor类型进行变换。py

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值