m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6291 fix bug in connect_network_with_dataset when using pynative mode

Browse Source 
Merge pull request !6291 from wangnan39/fix_bug_in_connect_net_with_data_in_pynative
pull/6291/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
parent 353b8b81c3 929a4b1d99
commit c9799fa678
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
mindspore/train/dataset_helper.py
Unescape View File

@ -89,7 +89,7 @@ def connect_network_with_dataset(network, dataset_helper):
raise RuntimeError("Dataset should be connected with network only in sink mode.") raise RuntimeError("Dataset should be connected with network only in sink mode.")
if not hasattr(dataset, '__ME_INITED__') and context.get_context("device_target") == "Ascend" and \ if not hasattr(dataset, '__ME_INITED__') and context.get_context("device_target") == "Ascend" and \
context.get_context("mode") == context.GRAPH_MODE and not context.get_context("enable_ge"): not context.get_context("enable_ge"):
dataset.__ME_INITED__ = True dataset.__ME_INITED__ = True
dataset_types, dataset_shapes = dataset_helper.types_shapes() dataset_types, dataset_shapes = dataset_helper.types_shapes()
queue_name = dataset.__TRANSFER_DATASET__.queue_name queue_name = dataset.__TRANSFER_DATASET__.queue_name

Write Preview
Loading…
Cancel
Save