昇腾社区首页
中文
注册

算子网络测试(训练)

简介

算子部署到算子库(OPP)后,可通过运行包含自定义算子的TensorFlow网络,验证算子在网络中的运行结果是否正确。

网络测试会覆盖算子开发的所有交付件,包含实现文件,算子原型定义、算子信息库以及算子适配插件。本节仅对验证的方法做介绍。

进行自定义算子在网络中的运行验证方法:执行包含自定义算子的模型训练。

MindStudio环境

基于MindStudio进行算子开发的场景下,基于TensorFlow框架的算子网络测试可通过模型训练的方式进行验证。

MindStudio提供了“Ascend Training”工程,工程创建完成后会自动生成相关交付件模板及编译配置文件,用户只需准备数据集及训练脚本,即可通过前台界面启动训练工程。

训练过程中可在运行窗口实时查看运行日志信息。

“Ascend Training”工程的创建及执行操作请参见MindStudio 用户指南中的“模型训练”章节。

命令行环境

基于命令行环境进行算子开发的场景下,算子网络测试的方法如下:

昇腾AI处理器BS9SX1A AI处理器中在线训练包含自定义算子的TensorFlow网络。

训练脚本的执行操作可参见TensorFlow网络模型迁移和训练指南