pytorch
yuanlulu
做过嵌入式Linux开发、深度学习图像方向的算法和落地。擅长c++和python。
展开
-
将pytorch的pth文件固化为pt文件
上面代码保存了state_dict, 所以保存的文件中是不含模型结构的,固化时需要从代码构造网络结构。好在项目是完全开源,将原项目下的model目录拷贝过来就行。,这个项目提供了预训练模型,我想要将该模型固化,然后转换格式后在嵌入式端使用。另外不能忘记调用eval() 来固化参数。我参考了一个开源的人像语义分割项目。原创 2023-03-08 15:13:54 · 2448 阅读 · 1 评论 -
pytorch模型转换问题记录
我在forward()函数里把tensor转换为python数值来显示,调用torch.jit.trace就会报错。在forward()函数里所有的操作都用tensor就没问题了。简单来说,原因就是我保存模型直接用torch.save(), 没法在其它脚本中直接加载,除非那个脚本中也有nn.Module子类的定义。为了在开发板上使用模型,我用转换脚本加载了我生成的pytorch模型文件,但是却报错了。我需要把我的模型保存为TorchScript格式。这样保存的pt就可以在其它python里加载了。...原创 2022-08-17 20:40:06 · 3384 阅读 · 2 评论