使用MindConverter迁移模型定义脚本

查看源文件

工具概述

MindConverter是一款模型迁移工具,可将PyTorch(ONNX)或Tensorflow(PB)模型快速迁移到MindSpore框架下使用。模型文件(ONNX/PB)包含网络模型结构(network)与权重信息(weights),迁移后将生成MindSpore框架下的模型定义脚本(model.py)与权重文件(ckpt)。

mindconverter-overview

此外,本工具支持通过在PyTorch网络脚本中增加API(pytorch2mindspore)的方式,将PyTorch网络模型迁移到MindSpore框架下。

  • 由于战略调整,MindConverter从1.9.0开始将不再演进,官网文档及代码将逐步下架,请知悉。

  • 如对MindConverter项目感兴趣,请移步1.7.0版本(文档详见MindConverter 1.7.0)。

  • MindConverter当前仅维护1.6.0和1.7.0两个版本,后续维护工作也将逐步向1.7.0倾斜。