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
be returned. The data and label would be passed to the network and loss
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.
Configure pynative mode or CPU, the training process will be performed with
dataset not sink.


Loading…
Cancel
Save