tensorflow 获取模型所有参数总和数量的方法

  • Z时代
  • 2024-01-10
  • 分类：综合

实例如下所示：

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 获取模型所有参数总和数量的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。

以上是 tensorflow 获取模型所有参数总和数量的方法 的全部内容， 来源链接： utcz.com/z/319084.html

回到顶部