为什么torch::jit:: load ()只能加载torch::jit::save()生成的文件

Question

我想在android项目中加载一个Py火炬模型的权重和偏差**.pt。我使用pytorch移动和used库来解析模型。当我使用torch::load api时，它会导致链接错误：

headers/torch/csrc/api/include/torch/serialize.h:130:未定义引用对`torch::serialize::InputArchive::InputArchive() headers/torch/csrc/api/include/torch/serialize.h:131:的未定义引用对`torch::serialize::InputArchive::load_from()的headers/torch/csrc/api/include/torch/serialize.h:139:未定义引用torch::serialize::InputArchive::try_read()

.Then I使用torch::jit::load to，但得到了执行错误：

终止C10：：Error类型的异常：torch::jit::load()从torch.save()接收了一个文件，但是torch::jit::load()只能加载torch.jit.save()生成的文件

Answer 1

也许原因是这。C++前端api目前只支持加载jit保存的张量文件。因此，唯一的解决方案是torch.load，然后是python中的torch.jit.save。