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

!17882 enable graph kernel when training retinaface_resnet50 on GPU 

From: @looop5
Reviewed-by: @anyrenwei,@ckey_dou,@gaoxiong1,@c_34,@kingxian
Signed-off-by: @ckey_dou,@gaoxiong1,@c_34
This commit is contained in:
mindspore-ci-bot 2021-06-07 16:29:04 +08:00 committed by Gitee
parent ad5e03c856 5db63a703f
commit e07c12a45e
2 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
akg

@ -1 +1 @@
Subproject commit 5dbebd8613b97bc6723bf8b29ee5ab480dfd6110
Subproject commit 67533ceaa0f0d2e5f9728c25b8c290ec15b56ef4

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

@ -33,6 +33,9 @@ from src.lr_schedule import adjust_learning_rate
def train(cfg):
context.set_context(mode=context.GRAPH_MODE, device_target='GPU', save_graphs=False)
if context.get_context("device_target") == "GPU":
# Enable graph kernel
context.set_context(enable_graph_kernel=True, graph_kernel_flags="--enable_parallel_fusion")
if cfg['ngpu'] > 1:
init("nccl")
context.set_auto_parallel_context(device_num=get_group_size(), parallel_mode=ParallelMode.DATA_PARALLEL,