pgjmblvch
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!10430 ascend support ps cache sink_size as 1 

From: @limingqi107
Reviewed-by: @chujinjin,@kisnwang
Signed-off-by: @kisnwang
This commit is contained in:
mindspore-ci-bot 2020-12-24 16:46:19 +08:00 committed by Gitee
parent 35520e6b38 37ffca4df8
commit 162e33baf0
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/train/dataset_helper.py
View File

@ -99,7 +99,8 @@ def connect_network_with_dataset(network, dataset_helper):
if (hasattr(dataset_iter, "sink_size") and dataset_iter.sink_size == 1) \
and (hasattr(dataset_iter, "sink_count") and dataset_iter.sink_count == 1) \
and context.get_context("device_target") == "Ascend" \
and context.get_context("mode") == context.GRAPH_MODE:
and context.get_context("mode") == context.GRAPH_MODE \
and ms_role != "MS_WORKER":
if not hasattr(dataset, '__network__'):
dataset.__network__ = network