diff --git a/mindspore/nn/cell.py b/mindspore/nn/cell.py index 97e0466b2a..75848bd7d7 100755 --- a/mindspore/nn/cell.py +++ b/mindspore/nn/cell.py @@ -54,6 +54,8 @@ class Cell(Cell_): Args: auto_prefix (bool): Recursively generate namespaces. Default: True. + flags (dict): Network configuration information, currently it is used for the binding of network and dataset. + Users can also customize network attributes by this parameter. Default: None. Supported Platforms: ``Ascend`` ``GPU`` ``CPU``