使用迁移工具

使用迁移工具进行模型迁移的核心步骤如下：

1. 进入迁移工具所在路径。

   cd Ascend-cann-toolkit安装目录/ascend-toolkit/latest/tools/ms_fmk_transplt/

2. 执行脚本迁移任务，参考表1配置信息。

   ./pytorch_gpu2npu.sh -i 原始脚本路径 -o 脚本迁移结果输出路径 -v 原始脚本框架版本 [-s] [-m] [distributed -t 目标模型变量名 -m 训练脚本的入口文件]

   命令示例：

   ./pytorch_gpu2npu.sh -i /home/train/ -o /home/out -v 1.11.0 [-s] [-m]

3. 迁移完成后的训练脚本可在NPU上运行，执行以下训练命令。
   如果训练正常进行，开始打印迭代日志，说明训练功能迁移成功。

    python main.py -a resnet50 -b 32 --gpu 1 /data/imagenet/   # /data/imagenet/需替换为用户实际使用的ImageNet数据集的路径

4. 成功保存权重，说明保存权重功能迁移成功。