torch.autograd.profiler.profile.export_chrome_trace#
- profile.export_chrome_trace(path, metadata=None, use_python_export=False, cuda_graph_annotations=None, graph_lanes='none', default_stream=7)[source]#
Export an EventList as a Chrome tracing tools file.
The checkpoint can be later loaded and inspected under
chrome://tracingURL.- Parameters:
path (str) – Path where the trace will be written.