迁移后训练 迁移完成后的训练脚本可在NPU上运行,操作步骤如下: 单击菜单栏“Run > Edit Configurations...”。 进入运行配置界面,参数设置如图1所示。单击“OK”。 选择迁移后的训练脚本。 指定要训练的网络resnet50。 图1 运行配置 单击,启动训练。 正常训练过程如图2所示。检查训练过程是否正常,loss是否收敛。 训练过程中如遇protobuf相关报错,可根据报错提示安装相应版本的protobuf。 图2 训练过程截图 父主题: 分析迁移(PyTorch GPU2Ascend)