当前位置:   article > 正文

如何将 PyTorch 模型移植到 Tensorflow-第 1 部分

如何将 PyTorch 模型移植到 Tensorflow-第 1 部分

假设你有一个 PyTorch 模型,首先构建两个 python 脚本。一个是 PyTorch .py,另一个是 Tensorflow .py。

  1. Pytorch模型:yourmodel.pth
  2. Tensorflow模型:yourmodel.ckpt

参数名称

此移植任务涉及两件事:参数及其名称映射。

从参数名称开始,你必须知道Pytorch和tensorflow op名称之间的1-1映射规则。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/404507
推荐阅读
相关标签
  

闽ICP备14008679号