tensorflow 模型保存和加载

 

 使用 tf.train.Saver

• 保存：tf.train.Saver.save(sess, save_path, global_step=None, latest_filename=None, meta_graph_suffix=\'meta\', write_meta_graph=True, write_state=True)
• 加载：tf.train.Saver.restore(sess,save_path)

步骤为：定义输入 placeholder

              定义graph

              定义 loss

              定义 optimizer, train=optimizer.minmize(loss)

              定义Session, Session.run(train)

              保存结果 tf.train.Saver.save(sess,path)