From cba92f7ea18dec2f5ff484a664140b22d00bf4f3 Mon Sep 17 00:00:00 2001 From: "wangnan39@huawei.com" Date: Fri, 26 Mar 2021 16:27:06 +0800 Subject: [PATCH] add platform for connect_network_with_dataset --- mindspore/train/dataset_helper.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/mindspore/train/dataset_helper.py b/mindspore/train/dataset_helper.py index 6ca4098937..e7618d8eb0 100644 --- a/mindspore/train/dataset_helper.py +++ b/mindspore/train/dataset_helper.py @@ -102,6 +102,9 @@ def connect_network_with_dataset(network, dataset_helper): Cell, a new network wrapped with 'GetNext' in the case of running the task on Ascend in graph mode, otherwise it is the input network. + Supported Platforms: + ``Ascend`` ``GPU`` + Examples: >>> from mindspore import DatasetHelper >>>