forked from mindspore-Ecosystem/mindspore
fix epoch repeat doc problem
This commit is contained in:
parent
a44d734741
commit
732dd01a6a
|
@ -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).
|
||||
|
|
|
@ -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.
|
||||
|
|
Loading…
Reference in New Issue