tensorflow初探---模型文件保存和加載

tensorflow模型文件能夠大體分爲兩種：

1.經過tensorflow.train.Saver來生成

1)保存

saver = tf.train.Saver(tf.all_variables())
saver.save(sess, 'model.ckpt')

2)加載

saver=tf.train.Saver(tf.all_variables())
save.restore(sess, 'model.ckpt')

2.經過tensorflow.GraphDef

1)保存

tf.train.write_graph(sess.graph_def, path, filename)
或者
witf tf.gfile.FastGFile(filename, 'wb') as f:
    f.write(sess.graph_def.SerializeToString())

2)加載

with tf.gfile.FastGFile(filename, 'rb') as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())
    tf.import_graph_def(graph_def, name='')

暫時寫這些，未完待續