saved_modelのグラフの中身がどうなってるかを確認する

1. ログにグラフ情報を書き出す

import tensorflow as tf
log_dir = "/tmp/tf-log/"

graph = tf.Graph()
with tf.Session(graph=graph) as session:
    loaded = tf.saved_model.load(
        session,
        {'serve'},
        ここにsaved_modelのpath
    )
    
    train_writer = tf.summary.FileWriter(log_dir)
    train_writer.add_graph(session.graph)

2. tensorboardで見る

tensorboard --logdir /tmp/tf-log

追記(2020/05/04): netronつかったらguiでグラフ構造確認できるわ...
github.com