參數(shù)初始化：

sess = tf.InteractiveSession() sess.run(tf.global_variables_initializer()) # 初始化模型的參數(shù)

參數(shù)保存：

saver = tf.train.Saver() save_path = saver.save(sess, "E:\\projects\\PycharmProjects\\OCR\\model\\") print(save_path)

在訓(xùn)練循環(huán)中，定期調(diào)用 saver.save() 方法，向文件夾中寫(xiě)入包含了當(dāng)前模型中所有可訓(xùn)練變量的 checkpoint 文件。
saver.save(sess, FLAGS.train_dir, global_step=step)
global_step是訓(xùn)練的第幾步

保存參數(shù)：

import tensorflow as tfW = tf.Variable([[1, 2, 3]], dtype=tf.float32) b = tf.Variable([[1]], dtype=tf.float32)saver = tf.train.Saver() sess = tf.InteractiveSession() tf.global_variables_initializer().run() # 必須要指定文件夾，保存到ckpt文件 save_path = saver.save(sess, "winycg/1.ckpt") print(save_path)

參考博文： https://blog.csdn.net/winycg/article/details/78572438

總結(jié)

以上是生活随笔為你收集整理的DeepLearning：tensorflow 参数初始化和参数保存的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。