tensorflow模型参数如何查看

tensorflow

1455

2024/4/20 14:19:03

栏目: 深度学习

在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(10, activation='softmax')
])

model.summary()

执行以上代码将会输出模型的参数信息，包括每一层的名称、类型、输出shape以及可训练参数的数量。

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

tensorflow模型参数如何查看

最新知识库

相关标签