How to visualize model graph in weights and biases


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.

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.

This topic was automatically closed 60 days after the last reply. New replies are no longer allowed.