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

前端 未结 4 785
一向
一向 2020-12-13 13:51

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 t

4条回答
  •  感动是毒
    2020-12-13 14:15

    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/get_started/graph_viz

提交回复
热议问题