mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

adjust format of api

This commit is contained in:
sl_wang 2021-06-28 19:43:17 +08:00
parent 19e2e063c9
commit d91bf52e09
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/train/train_thor/convert_utils.py
View File

@ -133,14 +133,15 @@ class ConvertNetUtils():
This interface is used to convert a network to thor layer network, in order to calculate and store the
second-order information matrix.
Notes:
Note:
This interface is automatically called by the second-order optimizer thor.
Args:
net (Cell): network to be trained by the second-order optimizer thor.
net (Cell): Network to be trained by the second-order optimizer thor.
Examples:
>>> ConvertNetUtils().convert_to_thor_net(net)
"""
net.update_cell_prefix()