m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add description about sink_size

Browse Source
pull/3556/head
jinyaohui 5 years ago
parent 9c461f5565
commit ac6d3d557c
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File

4
mindspore/train/model.py
Unescape View File

@ -523,7 +523,9 @@ class Model:
of data will be transferred one by one. The limitation of data transmission per time is 256M. of data will be transferred one by one. The limitation of data transmission per time is 256M.
Args: Args:
epoch (int): Total number of iterations on the data. epoch (int): Generally, total number of iterations on the data per epoch.
When dataset_sink_mode is set to true and sink_size>0, each epoch sink sink_size
steps on the data instead of total number of iterations.
train_dataset (Dataset): A training dataset iterator. If there is no train_dataset (Dataset): A training dataset iterator. If there is no
loss_fn, a tuple with multiply data (data1, data2, data3, ...) should be loss_fn, a tuple with multiply data (data1, data2, data3, ...) should be
returned and passed to the network. Otherwise, a tuple (data, label) should returned and passed to the network. Otherwise, a tuple (data, label) should

Write Preview
Loading…
Cancel
Save