def __init__(self,
enable_profiling=False,
profiling_options=None
)
ProfilingConfig类的构造函数,用于配置Profiling功能。
无。
参数名 |
输入/输出 |
描述 |
---|---|---|
enable_profiling |
输入 |
是否开启Profiling功能。
|
profiling_options |
输入 |
Profiling配置选项。
说明:
配置示例:
profiling_options = '{"output":"/tmp/profiling","training_trace":"on",task_trace":"on","fp_point":"","bp_point":"","aic_metrics":"PipeUtilization"}' |
返回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) |