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’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