tensorflow模型参数怎么查看

   2025-02-07 8400
核心提示:在TensorFlow中,可以使用model.summary()方法来查看模型的参数。这个方法会打印出模型的结构以及每一层的参数数量。示例如下:i

在TensorFlow中,可以使用model.summary()方法来查看模型的参数。这个方法会打印出模型的结构以及每一层的参数数量。示例如下:

import tensorflow as tf# 创建模型model = tf.keras.Sequential([    tf.keras.layers.Dense(64, activation='relu', input_shape=(784,)),    tf.keras.layers.Dense(64, activation='relu'),    tf.keras.layers.Dense(10, activation='softmax')])# 打印模型参数model.summary()

运行上述代码,会输出模型的结构以及每一层的参数数量。例如:

Model: "sequential"_________________________________________________________________Layer (type)                 Output Shape              Param #   =================================================================dense (Dense)                (None, 64)                50240     _________________________________________________________________dense_1 (Dense)              (None, 64)                4160      _________________________________________________________________dense_2 (Dense)              (None, 10)                650       =================================================================Total params: 55,050Trainable params: 55,050Non-trainable params: 0_________________________________________________________________

在上述输出中,每一层的参数数量都会显示在Param #这一列中。

 
 
更多>同类维修知识
推荐图文
推荐维修知识
点击排行
网站首页  |  关于我们  |  联系方式  |  用户协议  |  隐私政策  |  网站留言