有任何 pytorch 工具来监控神经网络的训练吗？答案

【问题标题】：Any pytorch tools to monitor neural network's training?有任何 pytorch 工具来监控神经网络的训练吗？
【发布时间】：2018-01-13 03:13:05
【问题描述】：

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

【问题讨论】：

标签： neural-network deep-learning pytorch tensorboard

【解决方案1】：

PyTorch 1.1.0 通过 torch.utils.tensorboard 原生支持 TensorBoard。 API 与 tensorboardX 非常相似。有关详细信息，请参阅documentation。

【讨论】：

【解决方案2】：

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

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

【讨论】：

【解决方案3】：

我以前在论坛上问过这个问题。 Tensorboard 对 Tensorflow 来说似乎非常方便，它也是库/框架本身的一部分。但是，PyTorch 不会采用相同的方法。但是有一个名为 visdom here 的库是由 Facebook 发布的，它可以帮助您记录培训信息。这为您提供了以您想要的方式记录信息的灵活性。虽然这意味着很大的灵活性，但也意味着您需要编写一些额外的代码才能使事情正常进行。

【讨论】：

【解决方案4】：

跟进 blckbird 的回答，我也是 Tensorboard-PyTorch 的忠实粉丝。但是我也发现它的 API 级别相对较低，我一遍又一遍地编写了很多类似的代码来进行日志记录。所以（无耻的插件）我在它上面写了一个小包，用最少的代码自动监控网络训练实验。希望其他人觉得它有帮助。 pytorch-monitor

【讨论】：

【解决方案5】：

Minetorch 在过去的 2 场 Kaggle 比赛中帮助了我很多。我认为它已经准备好供其他人使用。它具有内置的 tensorboard 或 matplotlib 支持。还有许多其他使工作变得简单的功能，包括：

记录器
支持张量板
Matplotlib（生成 png 到文件）
自动恢复训练
自动保存最佳模型
自定义挂钩点
...

它仍在开发中，因此非常欢迎任何问题或 PR :)

【讨论】：