for v in tf.global_variables():
if 'global_step' in v.name:
var2.append(v) # 重置global step 調(diào)整學(xué)習(xí)率重新訓(xùn)練
else:
var1.append(v)
var1 = tf.train.Saver(var1)
...
"""Restore Model"""
save_file = tf.train.latest_checkpoint(hparams_at.train_dir)
if save_file:
print(save_file)
step = int(save_file.split('ckpt-')[-1]) + 1
var1.restore(sess, save_file)
sess.run(tf.variables_initializer([model.global_step]))
print("Go on")
else:
step = 0
sess.run(tf.global_variables_initializer())
print("Begin")
TensorFlow 加載部分變量
最后編輯于 :
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
- 文/潘曉璐 我一進(jìn)店門拦坠,熙熙樓的掌柜王于貴愁眉苦臉地迎上來连躏,“玉大人,你說我怎么就攤上這事贪婉》粗啵” “怎么了?”我有些...
- 文/不壞的土叔 我叫張陵疲迂,是天一觀的道長。 經(jīng)常有香客問我莫湘,道長尤蒿,這世上最難降的妖魔是什么? 我笑而不...
- 正文 為了忘掉前任幅垮,我火速辦了婚禮腰池,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己示弓,他們只是感情好讳侨,可當(dāng)我...
- 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著奏属,像睡著了一般跨跨。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上囱皿,一...
- 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼分苇!你這毒婦竟也來了慰技?” 一聲冷哼從身側(cè)響起,我...
- 正文 年R本政府宣布今野,位于F島的核電站,受9級特大地震影響罐农,放射性物質(zhì)發(fā)生泄漏条霜。R本人自食惡果不足惜,卻給世界環(huán)境...
- 文/蒙蒙 一涵亏、第九天 我趴在偏房一處隱蔽的房頂上張望宰睡。 院中可真熱鬧蒲凶,春花似錦、人聲如沸拆内。這莊子的主人今日做“春日...
- 文/蒼蘭香墨 我抬頭看了看天上的太陽麸恍。三九已至灵巧,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間或南,已是汗流浹背孩等。 一陣腳步聲響...