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

数据挖掘 深度学习 张量流 火炬
2021-09-20 14:58:31

是否有任何工具可以监控 PyTorch 中的网络训练?

我正在寻找与张量流中的张量板等效的东西。

2个回答

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

还有其他软件实现了 TensorBoard 的 PyTorch(和其他语言/框架)的包装器。据我所知,它们支持的功能较少。但是看看CrayonTensorboard-Logger

我也在使用Tensorboard-PyTorch (TensorboardX)。总的来说,我对此感到非常满意。

但不要试图可视化图表。至少没有一个有点复杂(例如 resnet50 不起作用)。他们的 github 页面上有一些关于它的问题

但最好经常查看 Pytorch 论坛。Pytorch 似乎移动得相当快。并且还计划了直接的张量板实现