python - 在 tensorflow 中关闭 session 不会重置图形

Question

当前图中可用的节点数在每次迭代中不断增加。这似乎不直观，因为 session 已关闭，并且它的所有资源都应该被释放。之前的节点在创建新session的时候还在徘徊是什么原因呢？这是我的代码:

for i in range(3):
    var = tf.Variable(0)
    sess = tf.Session(config=tf.ConfigProto())
    with sess.as_default():
        tf.global_variables_initializer().run()
        print(len(sess.graph._nodes_by_name.keys()))
    sess.close() 

输出:

5
10
15

Answer 1

关闭 session 不会按设计重置图形。如果你想重置图表，你可以像这样调用 tf.reset_default_graph()

for _ in range(3):
    tf.reset_default_graph()
    var = tf.Variable(0)
    with tf.Session() as session:
        session.run(tf.global_variables_initializer())
        print(len(session.graph._nodes_by_name.keys()))

或者你可以这样做

for _ in range(3):
    with tf.Graph().as_default() as graph:
        var = tf.Variable(0)
        with tf.Session() as session:
            session.run(tf.global_variables_initializer())
            print(len(graph._nodes_by_name.keys()))