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

!16952 enable graph_kernel while train inceptionv3 on gpu 

From: @yiyanzhi_akane
Reviewed-by: @anyrenwei,@ckey_dou
Signed-off-by: @ckey_dou
This commit is contained in:
mindspore-ci-bot 2021-05-26 16:12:48 +08:00 committed by Gitee
parent 2acc57583e 8ab4a0deca
commit 293eb7bb01
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
model_zoo/official/cv/inceptionv3/train.py
View File

@ -60,6 +60,9 @@ if __name__ == '__main__':
cfg = CFG_DICT[args_opt.platform] cfg = CFG_DICT[args_opt.platform]
create_dataset = DS_DICT[cfg.ds_type] create_dataset = DS_DICT[cfg.ds_type]
if args_opt.platform == "GPU":
context.set_context(enable_graph_kernel=True)
context.set_context(mode=context.GRAPH_MODE, device_target=args_opt.platform, save_graphs=False) context.set_context(mode=context.GRAPH_MODE, device_target=args_opt.platform, save_graphs=False)
if os.getenv('DEVICE_ID', "not_set").isdigit(): if os.getenv('DEVICE_ID', "not_set").isdigit():
context.set_context(device_id=int(os.getenv('DEVICE_ID'))) context.set_context(device_id=int(os.getenv('DEVICE_ID')))