Are there any tools to monitor network's training in PyTorch?
I am looking for an equivalent to tensorboard in tensorflow.
Data Science Stack Exchange is a question and answer site for Data science professionals, Machine Learning specialists, and those interested in learning more about the field. It only takes a minute to sign up.Sign up to join this community
I am using tensorboardX. It supports most of the features of TensorBoard. I am using the Scalar, Images, Distributions, Histograms and Text. Haven't tried the rest like audio and graph. But the repo also contains examples for those usecases. Installation can be done easily with pip. It's all explained in the readme.
There are also other software which implement a wrapper for PyTorch (and other languages/frameworks) of TensorBoard. As far as I know, they support fewer functionalities. But have a look at Crayon and Tensorboard-Logger.
I am also using Tensorboard-PyTorch (TensorboardX). Over all I am quite happy with it.
But don't try to visualize graphs. At least none with a bit of complexity (e.g. a resnet50 won't work). There are some issues about it on their github page.
But better check out the Pytorch forum frequently. Pytorch seems to move quite fast. And a direct tensorboard implementation is planned, as well.