Writer.add_graph(model)

I’m trying to pass the model to tensorboard.
Model takes multiple inputs such as input_ids,attention,etc.

So, how exactly I should use writer.add_graph in my case to make it work

I have the same issue.

You just need to call your model with inputs (could be a random input to the model)
It’s equivalent of calling forward function