有没有Pytork工具来监控神经网络的训练?

PyTorch 1.1.0本机支持张力板 torch.utils.tensorboard API与tensorboardX非常相似。请参阅 documentation

我正在使用 tensorboardX .它支持张力板的大部分(如果不是全部)功能。我使用标量、图像、分布、直方图和文本。我没有试过其他的方法,比如音频和图形,但repo也包含了这些用例的示例。安装可以很容易地完成 pip

• Crayon
• Tensorboard-Logger

我以前在论坛上问过这个问题。Tensorboard对于Tensorflow来说似乎非常方便,它也是库/框架本身的一部分。然而,PyTorch不会采取同样的方法。但是有一个图书馆叫做 visdom here

继布莱克伯德的回答之后,我还是Tensorboard PyTorch的忠实粉丝。然而,我也发现它的API相对较低,我一遍又一遍地编写了许多类似的代码来进行日志记录。所以(无耻的插件)我在上面写了一个小软件包,用最少的代码自动化监控网络训练实验。希望其他人会觉得这很有帮助。 pytorch-monitor

Minetorch 在过去的2场卡格舞比赛中帮了我很多。我认为它可以供其他人使用。它支持内置tensorboard或matplotlib。以及许多其他使工作变得简单的功能,包括:

• 记录器
• 自动恢复训练
• 用于自定义的挂钩点
• ...