自动迁移

操作步骤

1. 在训练脚本（examples-main/imagenet/main.py文件）中导入自动迁移的库代码。

   import torch 
   import torch_npu 
   from torch_npu.contrib import transfer_to_npu   
   .....

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

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

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