是否有任何工具可以监控 PyTorch 中的网络训练?
我正在寻找与张量流中的张量板等效的东西。
是否有任何工具可以监控 PyTorch 中的网络训练?
我正在寻找与张量流中的张量板等效的东西。
我正在使用tensorboardX。它支持 TensorBoard 的大部分功能。我正在使用标量、图像、分布、直方图和文本。还没有尝试过其他的,比如音频和图形。但 repo 还包含这些用例的示例。使用 pip 可以轻松完成安装。这一切都在自述文件中进行了解释。
还有其他软件实现了 TensorBoard 的 PyTorch(和其他语言/框架)的包装器。据我所知,它们支持的功能较少。但是看看Crayon和Tensorboard-Logger。
我也在使用Tensorboard-PyTorch (TensorboardX)。总的来说,我对此感到非常满意。
但不要试图可视化图表。至少没有一个有点复杂(例如 resnet50 不起作用)。他们的 github 页面上有一些关于它的问题。
但最好经常查看 Pytorch 论坛。Pytorch 似乎移动得相当快。并且还计划了直接的张量板实现。