from functools import reduce
from operator import mul
def get_num_params():
num_params = 0
for variable in tf.trainable_variables():
shape = variable.get_shape()
num_params += reduce(mul, [dim.value for dim in shape], 1)
return num_params
tensorflow 获取模型所有参数总和数量
最新推荐文章于 2024-08-01 14:42:58 发布
本文介绍如何在TensorFlow中获取模型的所有参数,并计算其总和数量,这对于理解和优化模型大小至关重要。
摘要由CSDN通过智能技术生成