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)