Browse Source

!3653 add epoch_num description master

Merge pull request !3653 from panfengfeng/add_epoch_num_description_master
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
e6c596a9d9
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/train/dataset_helper.py

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

@@ -52,6 +52,7 @@ class DatasetHelper:
sink_size (int): Control the amount of data each sink.
If sink_size=-1, sink the complete dataset each epoch.
If sink_size>0, sink sink_size data each epoch. Default: -1.
epoch_num (int): Control the number of epoch data to send.

Examples:
>>> dataset_helper = DatasetHelper(dataset)


Loading…
Cancel
Save