Browse Source

add platform for connect_network_with_dataset

pull/14187/head
wangnan39@huawei.com 4 years ago
parent
commit
cba92f7ea1
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/train/dataset_helper.py

+ 3
- 0
mindspore/train/dataset_helper.py View File

@@ -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
>>>


Loading…
Cancel
Save