我正在尝试将GAN与其他损失函数一起实现为我的模型的损失函数。在tensorflow后端的Keras中实现。 代码:
def discrim(input_shape):
img = Input(input_shape)
...
output = Dense(units=1)
return Model(inputs=img, outputs=output)
def combined(input_shape):
lr_img = Input(input_shape)
discrim = discrim(input_shape)
img_estimate = Sequential([...])(lr_img)
fake_score = discrim(img_estimate)
return Model(inputs=lr_img, outputs=[img_estimate, fake_score])
我只训练组合模型,以便在组合训练过程中对鉴别器进行训练。
我想使用Adam(lr = 0.1)作为组合模型的优化器,但是要使用Adam(lr = 0.01)作为Discrim模型的优化器。
谢谢。