libtorch模型部署

Question

上述为模型中部分自定义模块的python、C++实现，如何使用libtorch部署模型，如何才能让C++正确识别python导出的.pt模型中这些自定义模块呢?请教一下具体的流程

Answer 1

你保存模型的时候把网络结构一起保存下来无法读取吗？按理来说直接带网络结构就不需要在cpp下面定义网络结构的。
另外，cpp部署的话很少直接读取pt或者pth文件，都是转到onnx或者TorchScript，按照你的libtorch方式部署，我更建议你转成torch.jit.ScriptModule
https://zhuanlan.zhihu.com/p/486914187