mindspore.train.ConvertNetUtils

class mindspore.train.ConvertNetUtils[源代码]

将网络转换为thor层网络，用于计算并存储二阶信息矩阵。

convert_to_thor_net(net)[源代码]

该接口用于将网络转换为thor层网络，用于计算并存储二阶信息矩阵。

说明

此接口由二阶优化器thor自动调用。

参数：

  • net (Cell) - 由二阶优化器thor训练的网络。

支持平台：

Ascend GPU

样例：

>>> ConvertNetUtils().convert_to_thor_net(net)