How to Visualize the Model Graph of a PyG model in Tensorboard

[AnimeshSinha1309]

I have used both PyTorch and Tensorboard, and logging the model graph for any Graph Neural Network is being a problem everywhere, because the graphing function takes a dummy input, and that input must be a tensor, not a PyG Graph.
Any suggestions on how to subvert this and see my model graphs on Tensorboard.

[rusty1s]

Tensorboard should work out-of-the-box, but you should input tensors to your model rather than the data object. See here for an example.

[AnimeshSinha1309]

Thanks a lot, the mistake I was making was that I was passing in the data object into the model whereas I had to pass in tensors, after extracting from the data.