有什么pytorch工具可以监视神经网络的训练吗？

Question

是否有任何工具可以监控PyTorch中的网络培训？就像tensorflow中的tensorboard。

Answer 1

PyTorch 1.1.0 原生支持 TensorBoard 和torch.utils.tensorboard. API 与 tensorboardX 非常相似。有关更多详细信息，请参阅文档。

Answer 2

我正在使用tensorboardX。它支持 TensorBoard 的大部分（如果不是全部）功能。我正在使用标量、图像、分布、直方图和文本。我还没有尝试过其他的，比如音频和图表，但 repo 还包含这些用例的示例。安装可以很容易地完成pip。这一切都在 repo 的 README 文件中进行了解释。

还有其他 github 存储库为张量板实现了 PyTorch（和其他语言/框架）的包装器。据我所知，它们支持的功能较少。但是看看：

• 蜡笔
• Tensorboard-Logger