Save a block diagram of model in PyTorch

Asked Feb 07 '20 at 18:39

Active Feb 07 '20 at 18:39

Viewed 83 times

Keras has a nice functionality to save the model architecture as a block diagram

keras.utils.plot_model(self.model, plot_filepath.as_posix(), show_shapes=True)

Can this be achieved in PyTorch?

This answer talks about printing summary similar to Keras. Is there something similar to save the block diagram of model architecture?

asked Feb 07 '20 at 18:39

Nagabhushan S N

1

https://github.com/szagoruyko/pytorchviz – Reactgular Feb 07 '20 at 19:17
1

Does this answer your question? [How do I visualize a net in Pytorch?](https://stackoverflow.com/questions/52468956/how-do-i-visualize-a-net-in-pytorch) – akshayk07 Feb 07 '20 at 19:26

0 Answers0