向大佬请教一个问题,基于深度学习图像分类模型训练完成后,如何将其做成 LibTorch 调用的动态连接库,有没有那种教程可供参考呢?本人小白,望多多指教,虚心请教!
1
suifengingo OP 虚心请教 ing
|
2
sangleft Feb 24, 2021
没搞懂你的问题,保存的模型文件通过 libtorch 调用就好,编译成动态库的话我是用 C++包装了一下推理函数然后做成 dll 。而且 C++项目中也要有模型结构,还挺麻烦。
19 年做的,是要比 python 快一丢丢,现在可能有其他方式吧。 |