编辑TensorFlow检查点

时间:2016-10-30 01:28:01

标签: python tensorflow

我知道有一种方法可以检查TF检查点(如inspect_checkpoint.py中所述)。

但是,有没有办法更新此检查点中的某些权重并将其写回,而无需构建整个模型?例如,我想快速编辑其中一个图层的权重并将其写回,并且不想为此模型对应的深层网络编写代码。

1 个答案:

答案 0 :(得分:1)

看看MetaGraph saving and loading。那里有一些例子;您应该能够定义模型,保存MetaGraph,然后加载和修改它,而无需手动重新定义变量。