Tensorflow中變量保存與恢復_第1頁
Tensorflow中變量保存與恢復_第2頁
Tensorflow中變量保存與恢復_第3頁
全文預覽已結束

下載本文檔

版權說明:本文檔由用戶提供并上傳,收益歸屬內容提供方,若內容存在侵權,請進行舉報或認領

文檔簡介

1、Tensorflow 中變量保存與恢復SaverTensorflow 中,用 tensorflow.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)11 #例子import tensorflow as tf# 創建兩個變量v1= tf.Variable(tf.random_normal(784, 200, std

2、dev=0.35), name=v1)v2= tf.Variable(tf.zeros(200), name=v2)# 添加用于初始化變量的節點init_op = tf.global_variables_initializer()# Create a saver.saver = tf.train.Saver(tf.global_variables() # 運行,保存變量sess = tf.Session() saver.save(sess,my-model)12345678910111213141512345678 9101112131415Saver 可以使用提供的計數器自動為 checkp

3、oint 文件編號。這 使得在訓練模型時在不同的步驟保留多個檢查點。在 save() 方法中傳遞可選的 global_step 參數,可以對 checkpoint 文件 進行編號saver.save(sess, my-model, global_step=0) = filename: my-model-0saver.save(sess, my-model, global_step=1000) = filename: my-model-100012341234 #例子import tensorflow as tf# 創建兩個變量v1= tf.Variable(tf.random_normal(7

4、84, 200, stddev=0.35), name=v1)v2= tf.Variable(tf.zeros(200), name=v2) # 添加用于初始化變量的節點 init_op = tf.global_variables_initializer() # Create a saver.saver = tf.train.Saver(tf.global_variables() # 運行圖,打開會話,每 1000 次保存一個模型sess = tf.Session()for step in range(10000):sess.run(init_op)if step % 1000 = 0:sav

5、er.save(sess, base_path+my-model, global_step=step)1234567891011121314151617181234567891 01112131415161718運行結果: 恢復變量tf.train.Saver.restore(sess, save_path)11 #例子sess.run(tf.global_variables_initializer()saver = tf.train.Saver(tf.global_variables()#arg:獲取最近一次保存的變量文件名稱module_file = tf.train.latest_checkpoint(my-model) print(module_file)sav

溫馨提示

  • 1. 本站所有資源如無特殊說明,都需要本地電腦安裝OFFICE2007和PDF閱讀器。圖紙軟件為CAD,CAXA,PROE,UG,SolidWorks等.壓縮文件請下載最新的WinRAR軟件解壓。
  • 2. 本站的文檔不包含任何第三方提供的附件圖紙等,如果需要附件,請聯系上傳者。文件的所有權益歸上傳用戶所有。
  • 3. 本站RAR壓縮包中若帶圖紙,網頁內容里面會有圖紙預覽,若沒有圖紙預覽就沒有圖紙。
  • 4. 未經權益所有人同意不得將文件中的內容挪作商業或盈利用途。
  • 5. 人人文庫網僅提供信息存儲空間,僅對用戶上傳內容的表現方式做保護處理,對用戶上傳分享的文檔內容本身不做任何修改或編輯,并不能對任何下載內容負責。
  • 6. 下載文件中如有侵權或不適當內容,請與我們聯系,我們立即糾正。
  • 7. 本站不保證下載資源的準確性、安全性和完整性, 同時也不承擔用戶因使用這些下載資源對自己和他人造成任何形式的傷害或損失。

評論

0/150

提交評論