Linux 拨号vps windows公众号手机端

怎么查看tensorflow的参数量

lewis 5年前 (2020-04-19) 阅读数 10 #大数据
文章标签 tensorflow

要查看TensorFlow模型的参数量,可以使用以下代码片段:

import tensorflow as tf

model = tf.keras.applications.ResNet50() # 以ResNet50为例
model.summary()

运行这段代码后,会输出模型的结构信息,包括每一层的名称、输出形状和参数量等信息。可以根据参数量来大致估计整个模型的参数量。

版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

热门