diff --git a/mindspore/profiler/profiling.py b/mindspore/profiler/profiling.py index 6fb9cd98872dfb74e953f16982680b2e7a661dfb..9c5f533aa8427800bce565f923b6dcf2b87370eb 100644 --- a/mindspore/profiler/profiling.py +++ b/mindspore/profiler/profiling.py @@ -205,6 +205,9 @@ class Profiler: except (ProfilerIOException, ProfilerFileNotFoundException, RuntimeError) as err: logger.warning('Fail to write timeline data: %s', err) + os.environ['PROFILING_MODE'] = str("false") + context.set_context(enable_profiling=False) + def _analyse_step_trace(self, source_path, framework_parser): """ Analyse step trace data and save the result. @@ -275,11 +278,6 @@ class Profiler: timeline_analyser.write_timeline() timeline_analyser.write_timeline_summary() - def __del__(self): - """Disable the profiling collection service, called after training.""" - os.environ['PROFILING_MODE'] = str("false") - context.set_context(enable_profiling=False) - def _get_profiling_job_id(self): """Get profiling job id, which was generated by ada service.