mindspore.train.ConvertNetUtils

class mindspore.train.ConvertNetUtils[source]

Convert net to thor layer net, used to compute and store second-order information matrix.

convert_to_thor_net(net)[source]

This interface is used to convert a network to thor layer network, in order to calculate and store the second-order information matrix.

Note

This interface is automatically called by the second-order optimizer thor.

Parameters

net (Cell) – Network to be trained by the second-order optimizer thor.

Supported Platforms:

Ascend GPU

Examples

>>> ConvertNetUtils().convert_to_thor_net(net)