pytorch模型转caffe

    在做某些项目时，需要将训练好的模型部署到海思端，所以需要将模型转成caffe模型，根据这篇文章成功将模型转为了caffe模型（https://blog.csdn.net/qq_38964360/article/details/135755526）。
    另外，在转换的过程中我遇到了一些问题，在此记录一下，希望能给遇到相同问题的道友提供解决方法和思路。
    首先，将整个项目下载下来：git clone https://github.com/xxradon/PytorchToCaffe.git，然后导入自己的.pth模型文件，尝试运行：
    cd example
    python resnet_pytorch_2_caffe.py
    修改权重的路径，以及添加flatten层我就不说了，请参考上面的文章。运行的过程可能提示torch没有安装，或者版本不对，遇到这个问题可以在训练环境和linux环境下进行以下操作，统一  一下包的版本（我用的版本如下）：
    pip3 install torch==1.8.0
	pip3 install torchvision==0.9.0
   对了，记得把Caffe文件夹和pytorch_to_caffe.py复制并放在example目录下，如果不放的话，估计会提示找不到某些函数，像这样：
   ![请添加图片描述](https://i-blog.csdnimg.cn/direct/3962ac13b33f49c7984ae12066b3523a.png)