在TensorFlow中从已恢复的模型初始化变量

Question

假设我训练并保存以下简单网络（让我们称之为网络A）：

Input --> Conv1 --> Conv2 --> Conv3 --> Output

现在，我想恢复网络A，设计一个新的网络B并使用网络A的参数来初始化网络B的前3层：

Input --> Conv1 --> Conv2 --> Conv3 --> Conv4 --> Conv5 --> Output

如何在TensorFlow中执行此操作？

Answer 1

您可以创建网络A + B，并记住Conv3图层。然后，您可以计算Conv3输出的损失，然后训练一段时间。一旦您满意，只需切换到Conv5上的损失培训。