Browse Source

!6099 optimize the example in connect_network_with_dataset

Merge pull request !6099 from wangnan39/optimize_the_example
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
3d62ede8fb
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/train/dataset_helper.py

+ 1
- 1
mindspore/train/dataset_helper.py View File

@@ -61,7 +61,7 @@ def connect_network_with_dataset(network, dataset_helper):
Examples:
>>> # call create_dataset function to create a regular dataset, refer to mindspore.dataset
>>> train_dataset = create_dataset()
>>> dataset_helper = mindspore.DatasetHelper(train_dataset)
>>> dataset_helper = mindspore.DatasetHelper(train_dataset, dataset_sink_mode=True)
>>> net = Net()
>>> net_with_get_next = connect_network_with_dataset(net, dataset_helper)
"""


Loading…
Cancel
Save