p74368291
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

remove builtin profiling in model scaffolding

This commit is contained in:
chenhaozhe 2021-04-26 16:40:58 +08:00
parent 63c56099bf
commit 5ff36b9f7e
1 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
model_zoo/utils/model_scaffolding/src/moxing_adapter.py
View File

@ -18,7 +18,6 @@
import os import os
import functools import functools
from mindspore import context from mindspore import context
from mindspore.profiler import Profiler
from .config import config from .config import config
_global_sync_count = 0 _global_sync_count = 0
@ -102,14 +101,9 @@ def moxing_wrapper(pre_process=None, post_process=None):
if pre_process: if pre_process:
pre_process() pre_process()
if config.enable_profiling: # Run the main function
profiler = Profiler()
run_func(*args, **kwargs) run_func(*args, **kwargs)
if config.enable_profiling:
profiler.analyse()
# Upload data to train_url # Upload data to train_url
if config.enable_modelarts: if config.enable_modelarts:
if post_process: if post_process: