Any visualization tools for pytorch to help people debugging the network?

(N Hunter) #1

Hi, everyone, is there any tools like tensorboard for tensorflow that can plot the training loss, and weights etc. , to help people debugging the neuron network? Thanks.

(Masaki Kozuki) #2

I use tensorboardX, the wrapper for tensorboard.


Alternatively, you could use Visdom.

(N Hunter) #4

thanks. I will look into that

(N Hunter) #5

Thank you, I played around a little bit, quite impressive.