Tf上保存變量

1.定義文件的保存路徑

ckpt_dir="./ckpt_dir"

??? ifnotos.path.exists(ckpt_dir):

??? os.makedirs(ckpt_dir)

2.定義一個全局變量

global_step=tf.Variable(0,name='global_step',trainable=False)

這個全局變量是保存文件和提取文件的標識橙困,比如我現(xiàn)在要load什么時候保存的變量

3.定義saver方法

saver=tf.train.Saver()

注意任何變量定義在saver前面的都會被保存纷宇,在其后面定義的都不會被保存

4.保存變量

注意看前面定義的變量global_step,第一步給這個變量更新值(epoch)像捶,然后再保存桩砰。所以這個變量是以后load哪個文件的依據(jù)

global_step.assign(i).eval()#set and update(eval) global_step with index, i

saver.save(sess, ckpt_dir+"/model.ckpt",global_step=global_step)

5.load變量

ckpt=tf.train.get_checkpoint_state(ckpt_dir)

if ckpt and ckpt.model_checkpoint_path:

??? print(ckpt.model_checkpoint_path)

??? saver.restore(sess, ckpt.model_checkpoint_path)#restore all variables

最后編輯于
?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
  • 序言:七十年代末硼莽,一起剝皮案震驚了整個濱河市懂鸵,隨后出現(xiàn)的幾起案子匆光,更是在濱河造成了極大的恐慌酿联,老刑警劉巖贞让,帶你破解...
    沈念sama閱讀 222,104評論 6 515
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件喳张,死亡現(xiàn)場離奇詭異蹲姐,居然都是意外死亡,警方通過查閱死者的電腦和手機忙厌,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 94,816評論 3 399
  • 文/潘曉璐 我一進店門哥放,熙熙樓的掌柜王于貴愁眉苦臉地迎上來爹土,“玉大人胀茵,你說我怎么就攤上這事琼娘。” “怎么了瞒瘸?”我有些...
    開封第一講書人閱讀 168,697評論 0 360
  • 文/不壞的土叔 我叫張陵,是天一觀的道長赌蔑。 經(jīng)常有香客問我娃惯,道長石景,這世上最難降的妖魔是什么拙吉? 我笑而不...
    開封第一講書人閱讀 59,836評論 1 298
  • 正文 為了忘掉前任往史,我火速辦了婚禮佛舱,結果婚禮上,老公的妹妹穿的比我還像新娘脖祈。我一直安慰自己盖高,他們只是感情好眼虱,可當我...
    茶點故事閱讀 68,851評論 6 397
  • 文/花漫 我一把揭開白布撞蚕。 她就那樣靜靜地躺著过牙,像睡著了一般抒和。 火紅的嫁衣襯著肌膚如雪摧莽。 梳的紋絲不亂的頭發(fā)上,一...
    開封第一講書人閱讀 52,441評論 1 310
  • 那天,我揣著相機與錄音石咬,去河邊找鬼卖哎。 笑死亏娜,一個胖子當著我的面吹牛维贺,可吹牛的內(nèi)容都是我干的。 我是一名探鬼主播虐秋,決...
    沈念sama閱讀 40,992評論 3 421
  • 文/蒼蘭香墨 我猛地睜開眼用押,長吁一口氣:“原來是場噩夢啊……” “哼起愈!你這毒婦竟也來了抬虽?” 一聲冷哼從身側(cè)響起阐污,我...
    開封第一講書人閱讀 39,899評論 0 276
  • 序言:老撾萬榮一對情侶失蹤笛辟,失蹤者是張志新(化名)和其女友劉穎手幢,沒想到半個月后,有當?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體跺涤,經(jīng)...
    沈念sama閱讀 46,457評論 1 318
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點故事閱讀 38,529評論 3 341
  • 正文 我和宋清朗相戀三年院刁,在試婚紗的時候發(fā)現(xiàn)自己被綠了退腥。 大學時的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片阅虫。...
    茶點故事閱讀 40,664評論 1 352
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖窝革,靈堂內(nèi)的尸體忽然破棺而出虐译,到底是詐尸還是另有隱情,我是刑警寧澤侮攀,帶...
    沈念sama閱讀 36,346評論 5 350
  • 正文 年R本政府宣布,位于F島的核電站供鸠,受9級特大地震影響楞捂,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜胶坠,卻給世界環(huán)境...
    茶點故事閱讀 42,025評論 3 334
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望帖蔓。 院中可真熱鬧塑娇,春花似錦埋酬、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,511評論 0 24
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至酬姆,卻和暖如春辞色,著一層夾襖步出監(jiān)牢的瞬間浮定,已是汗流浹背桦卒。 一陣腳步聲響...
    開封第一講書人閱讀 33,611評論 1 272
  • 我被黑心中介騙來泰國打工闸盔, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留迎吵,地道東北人击费。 一個月前我還...
    沈念sama閱讀 49,081評論 3 377
  • 正文 我出身青樓蔫巩,卻偏偏與公主長得像圆仔,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子个从,可洞房花燭夜當晚...
    茶點故事閱讀 45,675評論 2 359

推薦閱讀更多精彩內(nèi)容