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

!19035 adjust api format of thor 

Merge pull request !19035 from wangshuangling/code_docs_api
This commit is contained in:
i-robot 2021-06-28 12:20:54 +00:00 committed by Gitee
parent 9dee4ddbb2 d91bf52e09
commit 6a0b05c5aa
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
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 This interface is used to convert a network to thor layer network, in order to calculate and store the
second-order information matrix. second-order information matrix.
Notes: Note:
This interface is automatically called by the second-order optimizer thor. This interface is automatically called by the second-order optimizer thor.
Args: 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: Examples:
>>> ConvertNetUtils().convert_to_thor_net(net) >>> ConvertNetUtils().convert_to_thor_net(net)
""" """
net.update_cell_prefix() net.update_cell_prefix()