1. 程式人生 > >tensorflow中模型儲存:tf.train.Saver.save,tf.train.train.Saver.restore

tensorflow中模型儲存:tf.train.Saver.save,tf.train.train.Saver.restore

在模型的儲存中我遇到過一次坑:先留一下懸念,我們先說一下怎樣 儲存模型和怎樣匯入模型:

模型的儲存;程式碼如下:

首先需要申明一個Saver物件:

saver = tf.train.Saver

with tf.Session as sess:

    saver.save(sess,'save_path')

這裡save_path 就很容易出現:Parent dictionary is not exiting ; 大致意思是這樣的;

這裡很多種方法解決,

1:save_path 弄成:./path/filename.ckpt的形式,但是我的不成功,

2:第二種就是通過匯入os 模組解決:程式碼如下:

cwd = os.getcwd() #獲取當前檔案的系統路徑

save_path = './save_model/filename.ckpt'

os.path.join(cwd +save_path)

好了我們再來說說模型的取得,

saver.restore(sess,'saver_path')  #saver_path 包含 檔名稱;