有什么方法可以查看张量板图中的参数总数？

Question

我在张量流中有一个图表，我想看看它在张量板上的可训练参数的总数。

我已经训练了模型并保存了它，然后我启动了张量板，在那里我可以看到训练、图表等的详细信息……但我没有看到参数的总数。有没有办法做到这一点？

我看到过使用tf.trainable_variables() 计数的帖子，但是张量板中没有更简单的方法吗？

Answer 1

您可以使用 tf.summary.text 编写模型摘要并将其显示在 tensorboard 中，您可以在其中查看模型摘要中的所有详细信息，例如可训练和不可训练参数等。

查看this 答案以获得更多参考。