使用前必读

概述

X2MindSpore工具可将基于PyTorch和TensorFlow开发的模型及其训练脚本根据适配规则迁移为可基于MindSpore运行的代码,大幅度提高了脚本迁移速度,降低了开发者的工作量。

  • X2MindSpore工具支持对包括但不限于模型列表列出的模型进行迁移,迁移成功后可直接运行,部分模型需要根据实际情况进行少量适配。
  • X2MindSpore工具当前仅支持PyTorch和TensorFlow训练脚本的迁移。
  • X2MindSpore工具会在迁移后的目录中生成适配层文件目录x2ms_adapter。该目录中保存的是基于MindSpore实现的PyTorch/TensorFlow API替换接口,工具会将原PyTorch/TensorFlow API按照映射关系迁移至适配层API,迁移后代码依赖适配层运行。
  • X2MindSpore工具能支持模型列表中的工程迁移后训练成功且收敛,对最终精度和性能暂不作保证。
  • 通过X2MindSpore工具迁移后的训练工程支持在MindSpore1.7及以上版本上运行。

使用限制

环境准备

配置环境变量

  1. 以运行用户在任意目录下执行vi ~/.bashrc命令,打开.bashrc文件,在文件最后一行后面添加以下内容(以非root用户的默认安装路径为例)。
    # Ascend-cann-toolkit环境变量(请根据实际路径修改)
    source ~/Ascend/ascend-toolkit/set_env.sh
  2. 执行:wq!命令保存文件并退出。
  3. 执行source ~/.bashrc命令使其立即生效。