tensorflow ‘/biases/Adam_1’not in ckpt file

在加載模型的時候,有時候會遇到這個問題,提示在ckpt文件裏找不到Adam_1變量,在使用rmsprop或者adam優化器的時候,他們會在自己內部建立一個adam或者rmsprop變量,使用SGD優化器的時候不會新建變量,所以不會出現在這種問題,之前在我的博客裏簡要介紹了這個問題,不過和這個不太一樣
http://blog.csdn.net/qq_25737169/article/details/77856961
解決這個問題可以使用下面的方法:

temp = set(tf.global_variables())
train_op = tf.train.AdamOptimizer(lr).minimize(loss, vars_list)
sess.run(tf.initialize_variables(set(tf.global_variables())-temp))

至於選擇性加載網絡權重的方法可以見我另一篇博客
http://blog.csdn.net/qq_25737169/article/details/78125061

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章