Skip to content

Commit

Permalink
Update mmengine/hooks/profiler_hook.py
Browse files Browse the repository at this point in the history
  • Loading branch information
zhouzaida committed Dec 27, 2022
1 parent d41afb7 commit aebd381
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion mmengine/hooks/profiler_hook.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,8 @@ class ProfilerHook(Hook):
display or tensorboard display. The terminal display content can be
adjusted through 'EventList.table()'
from 'torch.autograd.profiler_util.py'.
If using tensorboard, save to '{work_dir}/tf_tracing_logs' by default.
If using tensorboard, save to '{work_dir}/tf_tracing_logs'
by default.
record_shapes (bool): Save information about operator's input shapes.
Defaults to False.
profile_memory (bool): Track tensor memory allocation/deallocation.
Expand Down

0 comments on commit aebd381

Please sign in to comment.