首页 > 解决方案 > 在 tensorflow 中组合计算图

问题描述

在 Tensorflow 中,我们目前正在使用具有三个网络的模型,即networkAnetworkBnetworkC。networkB和networkC冻结。

模型的流程是,

input --> networkA --> outputA --> networkB --> outputB
input and outputB --> networkC --> Loss

损失将用于反向传播过程来训练网络A

问题是networkAnetworkB在同一个图中,而networkC是不同的图。

如何组合图以反向传播损失以更新networkA中的梯度?

提前致谢....

标签: pythontensorflow

解决方案


推荐阅读