选择性地恢复TensorFlow中的权重

时间:2017-01-11 01:40:17

标签: tensorflow conv-neural-network

我有两个神经网络(TF计算图)定义如下:

  1. 模型1:一些输入和隐藏层+一个3路softmax层
  2. 模型2:一些输入和隐藏层(与模型1相同)+二进制Sigmoid层
  3. 模型1和模型2之间的唯一区别是输出层。

    我已经为模型1存储了一些经过训练的参数,并希望使用它们(输入和隐藏层)来初始化新模型2.理想情况下,对于模型2,输入和输入。隐藏图层应使用保存的权重,而输出图层将随机初始化。我想知道是否有可能在TensorFlow中实现这个部分权重恢复。这样做的最佳做法是什么?

    非常感谢!

0 个答案:

没有答案