Does mmdet have a profile to see how long each phase takes?

[vansin]

mmdet有profile吗，查看各个阶段耗时的 ?
Does mmdet have a profile to see how long each phase takes ?

[zachcoleman]

MMCV has a ProfilerHook see here

It uses the PyTorch profiler to show GPU utilization, dataloader time, and time spent in each operation. Here's an example of putting it in the config:

profiler = dict(
    type="ProfilerHook",
    by_epoch=False,
    profile_iters=10,
    on_trace_ready=dict(type="tb_trace", dir_name="hook_prof_logs"),
)

In the TensorBoard output (specified by setting type to betb_trace) you can see output like this:

Your other option is to just use Python profiling and gprof2dot to visualize them. This is messy and sometimes hard to visualize and doesn't correspond to all the backend C/C++ code being called or give you an idea of how the GPU is working: