libtorch问题:Unsupported value kind: Tensor

原因是你训练用的pytorch 可能是 1.2 ,后来运行测试在libtorch 上,libtorch 太新了

terminate called after throwing an instance of 'std::runtime_error'
  what():  The following operation failed in the TorchScript interpreter.
Traceback of TorchScript (most recent call last):
RuntimeError: Unsupported value kind: Tensor

在这里插入图片描述

解决办法:

换低版本的libtorch:

pytorch版本和libtorch版本对应上,有问题

原因是, 在训练过程中得到的 权重 weights 文件,是 在 pytorch 1.2 训练得到的;

而 pytorch 对应 不同的 libtorch 这个问题,之前一直没有意识到;

之前一直以为 是 cuda 或者 cxx ,c++ abi 的版本问题, 实际上不是;

现在验证的结果告诉我们:pytorch 1.8 和 pytorch 1.2 起码是兼容的(在tensor 支持上);pytorch 1.10.2 不和 pytorch 1.2 兼容;

在这里插入图片描述

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值