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

open graph_kernel flag in mobilenet v3

This commit is contained in:
zengzitao 2021-05-24 10:04:55 +08:00
parent 6bbbfe32a5
commit b5d08986cc
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
model_zoo/official/cv/mobilenetv3/train.py
View File

@ -160,6 +160,7 @@ if __name__ == '__main__':
config_ = None config_ = None
if args_opt.device_target == "GPU": if args_opt.device_target == "GPU":
config_ = config_gpu config_ = config_gpu
context.set_context(enable_graph_kernel=True)
elif args_opt.device_target == "CPU": elif args_opt.device_target == "CPU":
config_ = config_cpu config_ = config_cpu
else: else: