torch.jit.script模型转换为libtorch的心得

Facenet demo：https://www.cnblogs.com/muyisun/p/13338098.html

如果要转为libtorch可调用，应该使用torch.jit中的script或trace。

script会保留一些控制流，trace则会擦除。可以使用.code查看过程。

torch_trace_model = torch.jit.script(model, input)#script trace
print(torch_trace_model.code)

但是无论如何，pytorch都无法转第三方库。

https://zhuanlan.zhihu.com/p/96397421

 

总而言之，经过调研，facenet无法转为C++下可调用的libtorch模型，因为其包含了众多控制流和第三方库。

这给了我们一个启示：在设计网络模型时，应当只包含一些简单的控制流或仅输出模型，不应当有第三库参与。