温馨提示×

怎么查看tensorflow的参数量

tensorflow

小亿

133

2024-03-13 20:48:23

栏目: 深度学习

开发者测试专用服务器限时活动，0元免费领，库存有限，领完即止！点击查看>>

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

import tensorflow as tf

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

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

0 赞

0 踩

相关标签

产品服务

地区划分

专题活动

帮助支持

关于我们

售后咨询

7*24小时在线电话：400-100-2938

7*24小时在线 QQ：800811969

关注亿速云

亿速云公众号

手机网站二维码