How to visualize a network build by Pytorch

1

How to visualize a network build by Pytorch. I try tensorboard, but sometimes it fails because the support is not so good. Is there some nice tools to visualize the net?

2

You could try to export your model to ONNX and use this tutorial to visualize it.