如何在 pytorch 上可视化 CNN答案

【问题标题】：How do I visualize CNN on pytorch如何在 pytorch 上可视化 CNN
【发布时间】：2021-12-11 13:04:32
【问题描述】：

我刚刚了解了一点关于 pytorch 的知识。我用pytorch的官方文档搭建了一个CNN来计算各种优化算法的效果（我刚从SGD到adagrad）。但是大部分官方文档和教程视频都是在计算准确率和耗时的时候就结束了，模型可视化的代码，我完全不知道。想问一下类似下面两张图的可视化用的是什么。是matplotlib pyplot还是pytorch对应的可视化工具？enter image description here

【问题讨论】：

标签： pytorch conv-neural-network

【解决方案1】：

我不能告诉你使用什么库来生成你链接到的情节。有很多选项，一旦您拥有数据，您就可以使用所有这些选项。这些选项之一是 matplotlib。如果您想将绘图包含在 LaTeX 文档中，其他方法包括使用 Matlab 或 pgfplots。这些是我经常使用的工具。它们纯粹是主观的选择。

不过，pytorch 也支持 tensorboard，这对于实时跟踪训练进度特别有用。看看这个教程：https://pytorch.org/tutorials/recipes/recipes/tensorboard_with_pytorch.html

【讨论】：

【解决方案2】：

对我来说，它看起来像 matplotlib pyplot。在训练时，您可以存储所有损失值和准确度值，并在网络训练后绘制它们。

【讨论】：