使用 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)