Browse Source

modify train comment

tags/v1.2.0-rc1
changzherui 5 years ago
parent
commit
0e806c935d
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      mindspore/train/model.py

+ 2
- 1
mindspore/train/model.py View File

@@ -554,7 +554,8 @@ class Model:
returned and passed to the network. Otherwise, a tuple (data, label) should returned and passed to the network. Otherwise, a tuple (data, label) should
be returned. The data and label would be passed to the network and loss be returned. The data and label would be passed to the network and loss
function respectively. function respectively.
callbacks (list): List of callback objects which should be executed while training. Default: None.
callbacks (list, object): List of callback objects or callback object, which should be executed
while training. Default: None.
dataset_sink_mode (bool): Determines whether to pass the data through dataset channel. Default: True. dataset_sink_mode (bool): Determines whether to pass the data through dataset channel. Default: True.
Configure pynative mode or CPU, the training process will be performed with Configure pynative mode or CPU, the training process will be performed with
dataset not sink. dataset not sink.


Loading…
Cancel
Save