计算参数量

计算模型参数总数

原创于 2021-12-30 11:13:48 发布 · 303 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

该代码段定义了一个名为defnetParams的函数，用于计算给定模型的总参数数量。它遍历模型的所有参数，通过计算每个参数的大小（尺寸的乘积），累加得到总参数数。这在理解和优化深度学习模型的复杂性时非常有用。

def netParams(model):
    total_paramters = 0
    for parameter in model.parameters():
        i = len(parameter.size())
        p = 1
        for j in range(i):
            p *= parameter.size(j)
        total_paramters += p

    return total_paramters