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

ascend support ps cache sink_size as 1

This commit is contained in:
limingqi107 2020-12-24 12:37:11 +08:00
parent b42b9b6c80
commit 37ffca4df8
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