How to graph tf.keras model in Tensorflow-2.0?

后端 未结 4 1159
别那么骄傲
别那么骄傲 2020-12-14 19:11

I upgraded to Tensorflow 2.0 and there is no tf.summary.FileWriter(\"tf_graphs\", sess.graph). I was looking through some other StackOverflow questions on this

4条回答
  •  情歌与酒
    2020-12-14 20:01

    Another option is to use this website: https://lutzroeder.github.io/netron/

    which generate a graph with a .h5 or .tflite file.

    The github repo it's based on may be found here: https://github.com/lutzroeder/netron

提交回复
热议问题