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

Posted 2023-02-16

【中文标题】有啥方法可以查看张量板图中的参数总数？

【英文标题】：is there any way to see total number of parameters in tensorboard graph?有什么方法可以查看张量板图中的参数总数？

【发布时间】：2021-10-25 19:00:15

【问题描述】：

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

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

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

【参考方案1】：

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

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