ProfilingConfig构造函数
功能说明
ProfilingConfig类的构造函数,用于配置Profiling功能。
函数原型
def __init__(self, enable_profiling=False, profiling_options=None )
参数说明
返回值
返回ProfilingConfig类对象,作为NPURunConfig的参数传入。
约束说明
无。
调用示例
1 2 3 4 5 6 |
from npu_bridge.npu_init import * ... profiling_options = '{"output":"/home/HwHiAiUser/output","task_trace":"on"}' profiling_config = ProfilingConfig(enable_profiling=True, profiling_options= profiling_options) session_config=tf.ConfigProto(allow_soft_placement=True) config = NPURunConfig(profiling_config=profiling_config, session_config=session_config) |