自动迁移

操作步骤

修改内容少，只需在训练脚本中导入库代码，迁移后直接在昇腾NPU平台上运行（推荐）。

1. 在训练脚本main.py文件中导入自动迁移的库代码。

   from torch.utils.data import Subset
   import torch_npu 
   from torch_npu.contrib import transfer_to_npu   
   .....

2. 切换目录至迁移完成后的训练脚本所在路径（以/home/user为例），执行以下命令使用虚拟数据集进行训练，迁移完成后的训练脚本可在NPU上正常运行。
   开始打印迭代日志，说明训练功能迁移成功。

   cd /home/user
   python main.py -a resnet50 --gpu 1 --epochs 1 --dummy  # --gpu 1表示使用卡1，--epochs 1是指迭代次数为1

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