fix epoch repeat doc problem

This commit is contained in:
Yanjun Peng 2020-04-13 17:46:48 +08:00
parent a44d734741
commit 732dd01a6a
2 changed files with 5 additions and 0 deletions

View File

@ -394,6 +394,8 @@ class Dataset:
The order of using repeat and batch reflects the number of batches. Recommend that
repeat operation should be used after batch operation.
If dataset_sink_mode is False, here repeat operation is invalid.
If dataset_sink_mode is True, repeat count should be euqal to the epoch of training. Otherwise,
errors could occur since the amount of data is not the amount training requires.
Args:
count (int): Number of times the dataset should be repeated (default=None).

View File

@ -360,6 +360,9 @@ class Model:
Note:
CPU is not supported when dataset_sink_mode is true.
If dataset_sink_mode is True, epoch of training should be equal to the count of repeat
operation in dataset processing. Otherwise, errors could occur since the amount of data
is not the amount training requires.
Args:
epoch (int): Total number of iterations on the data.