How to plot model graph using keras plot_model function or model summary or anything similar?

TensorFlow General Discussion
, ,
1

I could not get the plot of model graph of tensorflow object detection models. It doesn’t even showup on tensorboard.
Is there any script to handle this? Where should I look for?

1 Like
2

Are you training/finetuning the model with model.fit?
Did you set theTensorboard callback?

3

Doing few-shot training from object detection model zoo on the custom dataset.

4

Do you have the Tensorboard callback in fit?

5

There is no fit. it is trained on SSDMetaArch custom model generated from config file.
It would be quick if I could share links but I can’t.

6

I answered a similar question on StackOverflow: python 3.x - How to graph tf.keras model in Tensorflow-2.0? - Stack Overflow

In short, you can try to load your model in a function decorated with tf.function, execute the forward pass, and now that the execution of the function has been traced trying to plot the graph on TensorBoard using tf.summary.trace_export

3 Likes