For example, I want to save the model graph to the tensorboard so that I can visualize it. Currently, I can only do this by overriding Trainer class, which is quite bothering to me. Is there any easy way to this?
Did you find an answer to your Question? I would also like to add the graph to the tensorboard.