昇腾社区首页
中文
注册

迁移后训练

迁移完成后的训练脚本可在NPU上运行,操作步骤如下:

  1. 单击菜单栏Run > Edit Configurations...
  2. 进入运行配置界面,参数设置如图1所示。单击“OK”
    1. 选择迁移后的训练脚本。
    2. 指定要训练的网络resnet50。
    图1 运行配置
  3. 单击,启动训练。

    正常训练过程如图2所示。检查训练过程是否正常,loss是否收敛。

    训练过程中如遇protobuf相关报错,可根据报错提示安装相应版本的protobuf。

    图2 训练过程截图