How to visualize model graph in weights and biases

[nikhilsalodkar]

Hi,

I have been trying to find an example on how to write a model graph in weights and biases just like we can do in Tensorboard using writer.add_graph() similar to as given here.
I am working with Pytorch lightning.

If someone can please refer me to a correct documentation or an example.

Thanks in advance.
Nikhil

[armanharutyunyan]

Hey Nikhil, if you have uploaded the .h5 file, you can click on the name of the file in the Files section of the run and that will open up Netron

[armanharutyunyan]

Hey @nikhilsalodkar, I wanted to follow up on this request. Please let us know if we can be of further assistance or if your issue has been resolved.