From ddfbf135c572fc5a904cd9aea0994048e962a9d0 Mon Sep 17 00:00:00 2001 From: co63oc Date: Mon, 7 Aug 2023 11:36:12 +0800 Subject: [PATCH] Update profiler.py (#56001) --- python/paddle/profiler/profiler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/profiler/profiler.py b/python/paddle/profiler/profiler.py index af50a0f27d0..2260382453c 100644 --- a/python/paddle/profiler/profiler.py +++ b/python/paddle/profiler/profiler.py @@ -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() -- GitLab