【发布时间】:2019-06-01 18:16:43
【问题描述】:
使用具有多个损失的 train_on_batch 训练 GAN 模型,我可以在编译模型时使用随机的loss_weights,还是有一些特定的策略来使用这些损失权重,如 Here 所述。在我的问题中,mean_sqaured_error 是 generated_image 和 original_image 的损失函数,binary_crossentropy 是 0 和 1 类的分类损失函数。
model.compile(optimizer=optimizer, loss=['mean_squared_error', 'binary_crossentropy'], loss_weights=[100,1])
【问题讨论】:
标签: tensorflow keras deep-learning loss-function generative-adversarial-network