Save model as png

vision
1

Can we store PyTorch deep learning model as a png image(Like Keras does)?

2

If you want to visualize the model, you can use netron to load PyTorch model: https://github.com/lutzroeder/netron

1 Like
3

Really helped me a lot
Thanks !!!