我现在正在尝试在keras中实现GAN。 当我训练网络时,我想同时使用GAN损耗和发电机损耗。 我可能会从这篇文章中得到一些帮助。在
它有点像“传递一致性损失的多尺度视频帧合成网络”中的损失函数: Loss function
仅含GAN损耗的原始代码如下:
self.generator = generator
self.discriminator = discriminator
self.gan = Sequential([generator, discriminator])
gen, dis, gendis = self.generator, self.discriminator, self.gan
gendis.compile(optimizer=opt, loss='binary_crossentropy')
我想把发电机损耗加在一起。因此,我尝试了以下方法:
^{pr2}$但它不起作用,并显示错误消息:“ValueError:Unknown entry in loss dictionary:”generator\u output“。只应输入以下键:['model_2']'。在
如何将发电机损耗添加到本培训程序中?在
非常感谢!在
您可能需要将loss dict的键设置为模型输出名称。所以如果第二个键应该是型号2,那么第一个可能是型号1?可以打印出模型摘要吗?在
相关问题 更多 >
编程相关推荐