pytorch保存的模型怎么才能用到TensorFlow里边啊

Question

pytorch保存的模型怎么才能用到TensorFlow里边啊？或者怎么保存的pytorch模型能用到TensorFlow里边啊

Answer 1

这个真的不好转的，像一下比较大的开源的项目可能会有这个东西，而一些比较常用的网络，类似VGG，ResNet啥的，基本上都是有对应的框架训练的，不会做转换。而其他一些像个人搭建的，你基本上只能自己摸索，我没实现过，但是看别人做的基本上也是需要根据torch网络搭建好tf网络，然后将模型权重赋值过去，一些特殊算子还得自己实现。
目前我遇到的开源项目里面，各种框架和模型之间互相转换的，最全的应该是yolov5了，给你找个转成tf的saved_model，下面还有pb格式和一些边缘计算格式的转换
https://github.com/ultralytics/yolov5/blob/6371de8879e7ad7ec5283e8b95cc6dd85d6a5e72/export.py#L30