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

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
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
文章知识点与官方知识档案匹配，可进一步学习相关知识
Python入门技能树人工智能深度学习416729 人正在系统学习中
注：本文转载自blog.csdn.net的热爱Coding的文章"http://blog.csdn.net/guotong1988/article/details/76802743"。版权归原作者所有，此博客不拥有其著作权，亦不承担相应法律责任。如有侵权，请联系我们删除。
复制链接
相关推荐
发表评论

评论记录：

未查询到任何数据！