From 929a4b1d99a18068b4ff35f64b29f0e4b95ee578 Mon Sep 17 00:00:00 2001 From: "wangnan39@huawei.com" Date: Tue, 15 Sep 2020 20:57:19 +0800 Subject: [PATCH] fix bug in connect_network_with_dataset --- mindspore/train/dataset_helper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/train/dataset_helper.py b/mindspore/train/dataset_helper.py index 2d3e2e7ce1..b4b774b49d 100644 --- a/mindspore/train/dataset_helper.py +++ b/mindspore/train/dataset_helper.py @@ -89,7 +89,7 @@ def connect_network_with_dataset(network, dataset_helper): 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 \ - 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_types, dataset_shapes = dataset_helper.types_shapes() queue_name = dataset.__TRANSFER_DATASET__.queue_name