保存模型:
saver=tf.train.saver()
saver.save(sess,path)#参数一个是会话session,一个是保存路径
这个不仅保存网络结构,还包括了参数。
载入模型:
saver=tf.train.saver()
saver.restore(sess,path)
载入保存的模型。
注意这几行代码是在with tf.Session() as sess:下
保存模型:
saver=tf.train.saver()
saver.save(sess,path)#参数一个是会话session,一个是保存路径
这个不仅保存网络结构,还包括了参数。
载入模型:
saver=tf.train.saver()
saver.restore(sess,path)
载入保存的模型。
注意这几行代码是在with tf.Session() as sess:下