add platform for connect_network_with_dataset

4 years ago · cba92f7ea1
parent d9f6a6277d
commit cba92f7ea1
1 changed files with 3 additions and 0 deletions
--- 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
        >>>