ProfilingConfig Constructor
Description
Constructs an object of class ProfilingConfig as the profiling configuration.
Prototype
def __init__(self, enable_profiling=False, profiling_options=None )
Options
Option |
Input/Output |
Description |
|---|---|---|
enable_profiling |
Input |
Profiling enable.
|
profiling_options |
Input |
Profiling options.
NOTE:
Example:
profiling_options =
'{"output":"/tmp/profiling","training_trace":"on",task_trace":"on","fp_point":"","bp_point":"","aic_metrics":"PipeUtilization"}'
|
Returns
An object of the ProfilingConfig class, as an argument passed to the NPURunConfig call.
Restrictions
None
Examples
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) |