任何用于监控神经网络训练的 pytorch 工具?

Any pytorch tools to monitor neural network's training?

PyTorch 中是否有监控网络训练的工具?就像tensorflow中的tensorboard。

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

还有其他 github 存储库实现了 PyTorch(和其他 languages/frameworks)到张量板的包装器。据我所知,它们支持的功能较少。但是看看:

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

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

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

  • 记录器
  • 支持 Tensorboard
  • Matplotlib(生成 png 文件)
  • 自动恢复训练
  • 自动保存最佳模型
  • 自定义挂钩点
  • ...

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

PyTorch 1.1.0 原生支持 TensorBoard torch.utils.tensorboard. The API is very similar to tensorboardX. See the documentation 了解更多详情。