From 08cd5b428d603a911d084f9c838538eac07afcdc Mon Sep 17 00:00:00 2001 From: kuizhiqing Date: Thu, 1 Sep 2022 11:54:27 +0000 Subject: [PATCH] fix todo for profiler --- fleetx/core/engine/eager_engine.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/fleetx/core/engine/eager_engine.py b/fleetx/core/engine/eager_engine.py index 64c670242..2db5cfde9 100644 --- a/fleetx/core/engine/eager_engine.py +++ b/fleetx/core/engine/eager_engine.py @@ -24,6 +24,7 @@ from paddle.distributed.sharding import group_sharded_parallel from paddle.fluid.dygraph.parallel import sync_params_buffers from paddle.distributed.fleet.utils.hybrid_parallel_util import fused_allreduce_gradients +from paddle.profiler import SummaryView sys.path.append("../../../") from fleetx.utils import logger @@ -222,8 +223,7 @@ def configure_optimizers(self): scheduler=scheduler, on_trace_ready=paddle.profiler.export_chrome_tracing( profiler_log), - #TODO(kzq) uncomment after bugfix - #record_shapes=record_shapes, + record_shapes=record_shapes, profile_memory=profile_memory) self.profiler.start() logger.warning( @@ -602,8 +602,6 @@ def inference(self, data): return self._inference_engine.predict(data) def _print_summary(self): - #TODO(kzq) move above - from paddle.profiler import SummaryView views_dict = { SummaryView.DeviceView: 'device', SummaryView.OverView: 'overview',