diff --git a/model_zoo/utils/model_scaffolding/src/moxing_adapter.py b/model_zoo/utils/model_scaffolding/src/moxing_adapter.py index 830d19a6fc9..25838a7da99 100644 --- a/model_zoo/utils/model_scaffolding/src/moxing_adapter.py +++ b/model_zoo/utils/model_scaffolding/src/moxing_adapter.py @@ -18,7 +18,6 @@ import os import functools from mindspore import context -from mindspore.profiler import Profiler from .config import config _global_sync_count = 0 @@ -102,14 +101,9 @@ def moxing_wrapper(pre_process=None, post_process=None): if pre_process: pre_process() - if config.enable_profiling: - profiler = Profiler() - + # Run the main function run_func(*args, **kwargs) - if config.enable_profiling: - profiler.analyse() - # Upload data to train_url if config.enable_modelarts: if post_process: