2016-10-30 44 views
0

我知道有一種方法可以檢查TF檢查點(如inspect_checkpoint.py中所述)。編輯TensorFlow檢查點

但是,有沒有辦法更新此檢查點中的某些權重並將其寫回,而無需構建整個模型?例如,我想快速編輯其中一個圖層的權重並將其寫回,並且不想爲此模型所對應的深層網絡編寫代碼。

回答

1

看看MetaGraph saving and loading。那裏有幾個例子。您應該能夠定義模型,保存MetaGraph,然後加載和修改它,而無需手動重新定義變量。

+0

謝謝!儘管理想的情況下,我認爲擁有'CheckpointWriter'方法會很棒(更多的討論[這裏](https://groups.google.com/a/tensorflow.org/forum/#!mydiscussions/discuss/_U7C1T_-XsI) )。無論如何,這似乎是目前最好的解決方案,所以我會接受它。 –