快速入门
本章节介绍TensorFlow训练场景下,如何快速进行AOE调优。如下以算子调优为例,如果需要进行子图调优,请在配置环境变量时,将AOE_MODE修改为1即可。
操作步骤
- 直接执行训练脚本,即可按照指定的调优模式自动调优。
- 查看调优结果。训练过程中,调优的关键日志信息如下。
1 2 3 4
# TFAdapter开启调优 in tune mode, training graph handled by tools # 工具启动调优 Aoe tuning graph.
调优完成后,生成如下文件。
- 自定义知识库:若满足自定义知识库生成条件(请参见图3),则会生成自定义知识库。生成的自定义知识库默认存储到${HOME}/Ascend/latest/data/aoe/custom/op/${soc_version}路径下。如何使用调优后的自定义知识库请参见如何使用调优后的自定义知识库。
- 调优结果文件:调优结束后,会在执行调优的工作目录下生成命名为“aoe_result_opat_${timestamp}_${pidxxx}.json”的文件,记录调优过程中被调优的算子信息。该文件中各字段解释请参见表2。
父主题: TensorFlow训练场景下在线调优