test_tensorflow.py 文件源码

def test_save_restore():
    tf.reset_default_graph()
    sess = tf.Session()
    path = '/tmp/tensor_saved_test2'
    meta_path = path + '.meta'

    r = tf.train.import_meta_graph(meta_path)
    r.restore(sess, path)

    s = _get_saver(100)
    s.save(sess, path)


    tf.reset_default_graph()
    sess2 = tf.Session()

    r2 = tf.train.import_meta_graph(meta_path)
    r2.restore(sess2, path)
    s2 = _get_saver(100)
    s2.save(sess2, path)