未验证 提交 ddfbf135 编写于 作者: C co63oc 提交者: GitHub

Update profiler.py (#56001)

上级 3aaea43c
......@@ -237,7 +237,7 @@ def export_chrome_tracing(
with profiler.Profiler(
targets=[profiler.ProfilerTarget.CPU, profiler.ProfilerTarget.GPU],
scheduler = (3, 10),
on_trace_ready=profiler.export_protobuf('./log')) as p:
on_trace_ready=profiler.export_chrome_tracing('./log')) as p:
for iter in range(10):
#train()
p.step()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册