-
fenglingsa
我用torch::gt函数得到了大于阈值的index,但是这个index在libtorch里怎么用呢?没办法像python一样使用[:][:][[index]的用法啊。
-
fenglingsa
@金天 尺寸应该没问题,直接在python中训练和推测都没问题,结果也是OK的。但是一旦trace到C++就报错,而且很有趣的是,如果只是单纯生成网络,不训练,仅有自生成的随机权重时,trace到C++没有问题,如果load训练后的参数,再trace,就会出现这个错误。
我查了一下pytorch的文档,发现官方文档里提过这个错误
![0_1569389525001_批注 2019-09-25 133116.jpg](正在上传 100%)
但是,我这边不存在原地更新张量的操作,我猜测这个错误可能与pytorch内部的内存操作有关,不知道是不是我网络权重参数过多造成的。 -
-
fenglingsa
如果不训练,直接用随机的参数生成C++脚本,没有问题。
把网络训练好后,再用torch.jit.trace输出成C++脚本时就会报错。