文档
注册

count_parameters

功能说明

统计模型参数量接口,根据用户提供的模型,统计模型参数量。

函数原型

count_parameters(model)

参数说明

参数名

输入/返回值

含义

使用限制

model

输入

待低秩分解模型。

必选。

数据类型:PyTorch或MindSpore模型。

调用示例

from ascend_utils.common.utils import count_parameters
print("Original model parameters:", count_parameters(model))
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词