下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

简介

功能说明

开发者在昇腾AI处理器上通过Estimator模式进行模型训练或在线推理时,可通过NPURunConfig类的构造函数,指定Estimator的运行配置。

NPURunConfig类继承了tf.estimator的RunConfig类,关于对RunConfig类原生接口的支持情况可参见RunConfig参数支持说明

函数原型

您可以在TensorFlow Adapter软件安装路径下的:python/site-packages/npu_bridge/estimator/npu/npu_config.py文件中查看NPURunConfig的原型定义,示例如下:

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
def __init__(self,
iterations_per_loop=1,
profiling_config=None,
model_dir=None,
tf_random_seed=None,
save_summary_steps=0,
save_checkpoints_steps=None,
save_checkpoints_secs=None,
...
)

NPURunConfig支持的详细参数请以后面章节的参数说明为准。

使用约束

使用多Device执行训练的场景下,不支持使用按时间保存文件的参数save_checkpoints_secs。

返回值

返回NPURunConfig类对象,作为NPUEstimator的初始化参数传入。

调用示例

NPURunConfig配置的通用使用方式如下所示:

1
2
3
4
5
6
from npu_bridge.npu_init import *
session_config=tf.ConfigProto()
config = NPURunConfig(
    session_config=session_config, 
    mix_compile_mode=False, 
    iterations_per_loop=1000)
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词