Docs icon indicating copy to clipboard operation
Docs copied to clipboard

Published 20 hours ago • verified publisher icon MegEngine

PyTorch / TF 模型权重转换样例

Open xxr3376 opened this issue 5 years ago • 0 comments

背景

  • https://github.com/MegEngine/MegEngine/issues/53 中有人询问如何从 PyTorch 已经训练好的模型转换到 MegEngine 上
  • 我们需要面临大量的现在 PyTorch / TensorFlow 模型的转换问题，而这个步骤并不那么显然

任务描述

  • 添加一篇文档，描述如何把现有的 PyTorch / TensorFlow 模型的参数转换到 MegEngine 上

目标

  • 在官网上，可以看到这篇文档
  • 其他框架的使用者可以按照文档内容，进行对应的权重转换操作

xxr3376 avatar Jul 03 '20 10:07 xxr3376