我正在尝试在卷积神经网络之后训练2种不同的GRU。但是,我想在第1阶段训练第一个GRU,然后在第2阶段训练第二个GRU,依此类推。但是,当我使用以下代码时:
predictions = tf.cond(train_gru_one, lambda: GRU_model_output1(layers[-1]), lambda: GRU_model_output2(layers[-1]))
我遇到以下错误:
ValueError: Initializer for variable cond_1/GRU_output1/Variable/ is from inside a control-flow construct, such as a loop or conditional. When creating a variable inside a loop or conditional, use a lambda as the initializer.
可以解决此错误吗?
非常感谢您的帮助!