PyTorch 到 Caffe 的模型转换工具

PyTorch入门实战教程

PyTorch2Caffe 是一个可以将 Pytorch 模型转换为 Caffe 模型的工具,支持多种网络结构(好像对upsampling支持还不太友好)。具体方法可以见下方代码实例:

工具地址:Github

PyTorch入门实战教程

Leave a Reply

Your email address will not be published. Required fields are marked *

返回顶部