1. 首页
  2. 数据库
  3. 其它
  4. PytorchToCaffe:从Pytorch模型到Caffe模型支持pytorch 0.3、0.3.1、0.4、0.4.1、1.0、1.0.1、1.2、1.3

PytorchToCaffe:从Pytorch模型到Caffe模型支持pytorch 0.3、0.3.1、0.4、0.4.1、1.0、1.0.1、1.2、1.3

上传者: 2021-02-22 10:43:16上传 ZIP文件 203.32KB 热度 11次
由于我们使用的pytorch模型基本上是动态图结构,因此动态图的问题在于,一旦前向不完整,就无法确定图结构,并且caffe是静态图框架,这将导致模型从pytorch转换为caffe到caffe。遇到很多问题,并且pytorch版本迭代非常快,因此不再建议使用此仓库。如果要将pytorch转换为caffe,建议通过此仓库使用pytorch-> onnx-> caffe 。 该代码主要来自 。感谢hahnyuan的贡献。 神经网络工具:转换器和分析器 提供用于pytorch和caffe的神经网络框架的工具。 nn_tools是根据MIT许可证发布的(有关详细信息,请参阅LICENSE文件)。
用户评论