Tensorflow:在一个网络中训练两个网络

时间:2018-05-18 22:38:35

标签: tensorflow

我在Tensorflow中面临以下问题:我构建了一个具有以下图层大小的网络:

input_0 = 100
input_1 = 1000
output = 10

整个网络看起来像:

[input_0, 300, 500, input_1, 800, 400, output]

现在我将数据提供给input_0并在之后运行优化步骤。在这里,我想使用直接的整个网络。但在这之后我还希望能够将一些数据提供给input_1并运行一个优化步骤,该步骤向后运行,但只到input_1层。这甚至可能吗?我的意思是应该有办法做到这一点。

简而言之:如果网络[input_0,...,output][input_1,...,output]在同一个图表中属于同一个网络时,如何相互独立地进行培训?

我尝试在Tensorflow中实现它,导致许多错误。我还尝试将网络分成两个网络。但我不知道如何正确连接它们。

有什么建议吗?

1 个答案:

答案 0 :(得分:1)

优化器接受var_list参数,只允许您更新一些权重。

请参阅GradientDescentOptimizer here的文档。