chenhaozhe
4 years ago
1 changed files with 1 additions and 7 deletions
Unified View
Diff Options
+ 1
- 7
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: | |||||
| profiler = Profiler() | |||||
| # Run the main function | |||||
| 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: | ||||