一般使用
saver.restore(sess, modeldir + "model.ckpt")
即可加載已經(jīng)訓(xùn)練好的網(wǎng)絡(luò)，可是有時(shí)候想值使用部分層的參數(shù)，這時(shí)候可以選擇在加載網(wǎng)絡(luò)之后重新初始化剩下的層
var_list = [weights['wd1'], weights['out'], biases['bd1'], biases['out'], global_step]initfc = tf.variables_initializer(var_list, name='init')
比如我們想從新初始化var_list中的各個(gè)層，在restore之后，再初始化即可
sess.run(init)saver.restore(sess, modeldir + "model.ckpt")print sess.run(global_step)#initialize several layersess.run(initfc)print sess.run(global_step)
即可發(fā)現(xiàn)部分變量重新初始化了
以上這篇基于tensorflow加載部分層的方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持VEVB武林網(wǎng)。
新聞熱點(diǎn)
疑難解答
圖片精選
