1

我想從這裏運行flowers_train.py腳本:https://github.com/tensorflow/models/tree/master/inception/inception微調以來網絡2倍(Tensorflow)

微調上的花朵數據集以來網絡。不同之處在於我想保存一個檢查點,然後再次運行flowers_train.py腳本,但現在恢復以前保存的檢查點。我注意到再次使用這個恢復器: restorer = tf.train.Saver(variables_to_restore)

在第一步中給了我很大的損失。那麼我需要使用restorer = tf.train.Saver()?

我也注意到提供的檢查點文件是434,9MB,但我保存的檢查點是389,9MB。

回答

0

什麼是variables_to_restore?如果它不包含最後一層,那麼是的,你會看到更高的損失和更小的文件大小。

+0

它包含最後一層,因爲我將fine_tune參數設置爲False。更具體地說,我正在接受1000個步驟的培訓,損失正在下降。當我想要恢復和訓練1000步時,損失會比前幾步更高,然後再次下降。我試圖使用恢復所有變量的普通恢復器= tf.train.Saver(),我認爲它起作用,因爲在第一步中丟失不再是高的。 – chrisrn

+0

這可能意味着最後一層沒有保存。 –

+0

Eitherway,tf.train.Saver()總是「最安全」的方式嗎?我的意思是使用這個恢復器不會出錯。 – chrisrn