Tensorboard Graph with custom training loop does not include my Model
问题 I have created my own loop as shown in the TF 2 migration guide here. I am currently able to see the graph for only the --- VISIBLE --- section of the code below. How do I make my model (defined in the ---NOT VISIBLE--- section) visible in tensorboard? If I was not using a custom training loop, I could have gone with the documented model.fit approach : model.fit(..., callbacks=[keras.callbacks.TensorBoard(log_dir=logdir)]) In TF 1, the approach used to be quite straightforward: tf.compat.v1