向训练有素的Tensorflow估算器添加一层

Question

我有一个使用Estimator API训练过的Tensorflow模型。现在我想加载此模型，在其中添加一个新层，然后仅训练新层（即冻结所有其他参数）。

最好的方法是什么？

我设法从检查点加载了模型并添加了新层，但是我不清楚如何将示例作为输入并进行训练。具体来说，我找不到输入的占位符。

Answer 1

我找到了一种简单的方法，并将其发布在这里，以防有人觉得有用：

1. 加载估算器检查点。
2. 在新名称范围下创建占位符和模型图的副本。
3. 提取两个范围内的所有可训练变量。
4. 为每个变量创建hasOwnProperty个操作。

---

代码：

assign