count_parameters
功能说明
统计模型参数量接口,根据用户提供的模型,统计模型参数量。
函数原型
count_parameters(network)
参数说明
参数名 |
输入/返回值 |
含义 |
使用限制 |
---|---|---|---|
network |
输入 |
待低秩分解模型。 |
必选。 数据类型:PyTorch或MindSpore模型。 |
调用示例
from ascend_utils.common.utils import count_parameters print("Original model parameters:", count_parameters(network))
父主题: 低秩分解接口