怎么查看tensorflow的参数量

avatar
作者
筋斗云
阅读量:1

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

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

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

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!