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

模型迁移流程总体主要包含以下操作,如表1所示:
迁移任务 |
说明 |
---|---|
如何将三方平台上的PyTorch模型代码映射到昇腾设备上。推荐使用自动迁移,通过添加from torch_npu.contrib import transfer_to_npu迁移代码,将三方平台上的代码在进行过程中自动映射为昇腾设备代码。 |
|
在昇腾设备上运行PyTorch模型代码时,必需的适配操作,包括训练前环境变量配置和模型脚本与启动脚本配置。 |
|
在昇腾设备上运行PyTorch模型代码时,可选的适配操作。用户需要查看特性是否适用于当前训练场景与设备,根据具体情况选择性适配。 |
|
完成脚本迁移后,在训练过程中,如果遇到问题,可以参考模型调试定位问题发生的代码位置。 |
|
适配完成后,用户可参考模型保存与导出选择模型的保存导出方式,此功能默认支持,无需特殊适配。 |
父主题: 模型迁移