昇腾社区首页
中文
注册

模型迁移流程

迁移适配流程如下图所示。

5.2-模型脚本迁移 5.3-环境变量和脚本配置 5.4-关键特性适配 5.5-模型调试 5.6-模型保存与导出 5.6-模型保存与导出 5.3-环境变量和脚本配置 5.4-关键特性适配

模型迁移流程总体主要包含以下操作,如表1所示:

表1 模型迁移流程

迁移任务

说明

模型脚本迁移

如何将三方平台上的PyTorch模型代码映射到昇腾设备上。推荐使用自动迁移,通过添加from torch_npu.contrib import transfer_to_npu迁移代码,将三方平台上的代码在进行过程中自动映射为昇腾设备代码。

环境变量和脚本配置

在昇腾设备上运行PyTorch模型代码时,必需的适配操作,包括训练前环境变量配置模型脚本与启动脚本配置

关键特性适配

在昇腾设备上运行PyTorch模型代码时,可选的适配操作。用户需要查看特性是否适用于当前训练场景与设备,根据具体情况选择性适配。

模型调试

完成脚本迁移后,在训练过程中,如果遇到问题,可以参考模型调试定位问题发生的代码位置。

模型保存与导出

适配完成后,用户可参考模型保存与导出选择模型的保存导出方式,此功能默认支持,无需特殊适配。