How to visualize model graph in weights and biases

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

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


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.