!3653 add epoch_num description master

Merge pull request !3653 from panfengfeng/add_epoch_num_description_master
5 years ago · e6c596a9d9
parent 6ea2aa4e73 4644085e92
commit e6c596a9d9
1 changed files with 1 additions and 0 deletions
--- a/mindspore/train/dataset_helper.py
+++ b/mindspore/train/dataset_helper.py
@ -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)