conv neural network – How do I visualize CNN on pytorch

I can not tell you what library is used to generate the plot you linked to.
There are plenty of options, all of which you can use once you have the data.
One of these options is matplotlib. Others include using Matlab or pgfplots if you want to include your plots in a LaTeX document. These are the tools I use somewhat frequently. They are purely subjective choices.

However, pytorch also supports tensorboard, which is especially useful for live tracking of the training progress.
Have a look at this tutorial: pytorch.org/tutorials/recipes/recipes/tensorboard_with_pytorch.html

Read more here: Source link