如何计算tensorflow 模型中可训练参数的总数?
0 231
1

是否有函数调用或其他方法来计算tensorflow 模型中的参数总数?

参数的意思是:可训练变量的N个dim向量有N个参数,NxM矩阵有N*M参数等。 因此,本质上,我想在tensorflow会话中对所有可训练变量的维数的乘积求和。

收藏
2021-02-06 10:58 更新 anna •  5042
共 1 个回答
高赞 时间
0

循环tf.trainable_variables()中的每个变量。

total_parameters = 0for variable in tf.trainable_variables():
    # shape is an array of tf.Dimension
    shape = variable.get_shape()
    print(shape)
    print(len(shape))
    variable_parameters = 1
    for dim in shape:
        print(dim)
        variable_parameters *= dim.value
    print(variable_parameters)
    total_parameters += variable_parameters
print(total_parameters)

Via:https://stackoverflow.com/a/38161314/14964791

收藏
2021-02-06 11:24 更新 karry •  4540