Search code examples
tensorflowkeras

Is there an easy way to get something like Keras model.summary in Tensorflow?


I have been working with Keras and really liked the model.summary() It gives a good overview of the size of the different layers and especially an overview of the number of parameters the model has.

Is there a similar function in Tensorflow? I could find nothing on Stackoverflow or the Tensorflow API documentation.


Solution

  • I haven't seen anything like model.summary() for the tensorflow... However, I don't think you need it. There is a TensorBoard, where you can easily check the architecture of the NN.

    https://www.tensorflow.org/tensorboard

    enter image description here https://www.tensorflow.org/get_started/graph_viz