昇腾社区首页
开发者
资源
支持
积分兑换 NEW

dump数据前处理

原始训练脚本处理（GPU）

进入训练脚本所在路径（如“~/models/official/resnet”）。

  • 编辑resnet_run_loop.py文件，修改如下（以下行数仅为示例，请以实际为准）：
    • 注释掉第83、85行

    • 注释掉第576~583行

    • 第596行，修改为“return None”。

  • 为控制训练时间，可修改相应参数，示例如下：

    编辑cifar10_main.py文件，将train_epochs的值改为1。

迁移后的训练脚本处理（NPU）

进入训练脚本所在路径（如“~/MindstudioProjects/MyTraining-tensorflow/models_msft/models_npu_20220927162439/official/resnet”）。

  • 编辑resnet_run_loop.py文件，修改如下（以下行数仅为示例，请以实际为准）：
    • 注释掉第83、85行

    • 注释掉第575~582行

    • 第595行，修改为“return None”。

  • 为控制训练时间，可修改相应参数，示例如下：
    • 编辑cifar10_main.py文件，将train_epochs的值改为1。

    • 编辑resnet_run_loop.py文件，把max_steps设置为1。