Keras鉴别器的不同学习率

时间:2018-10-23 16:13:20

标签: python keras

我正在尝试将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模型的优化器。

谢谢。

0 个答案:

没有答案