4.1 损失函数
GAN的训练优化目标其实就是如下函数:
可以看到,这里有两个loss:一个是训练鉴别器时使用的 D_loss, 另一个是训练生成器时使用的 G_loss。
(资料图)
而这个模型的目标是要最小化 G_loss, 以及最大化 D_loss。
这里我们使用了Adam优化策略和BCE loss 来优化这两个。 于是可以写出:
4.2 模型迭代
在模型迭代的过程中,我们会做如下步骤:
我们会读取图像和标签(暂时没用)
然后生成一个随机的噪声z 并放入生成器生成一张假的图片,称为fake_img
之后将fake _ image 放入鉴别器得出 fake _ image 的评分
将这个评分与 1 比较得到 G_loss
再将输入的图像和fake_image 加上真假标签后放入鉴别器中得到D _ loss
循环以上过程 opt.epoch 次
由此,我们可以得到这部分的代码:
至此,模型已经训练完毕。
这里我们使用 torchvision.utils 库中的 save_image函数来存储图片,用法如下:
我们使用torch.save来保存模型即其中的参数,实际上需要保存的其实就是 generator 和 discriminator 这两个东西,用法如下:
然后使用的时候就只需要load一下就行了:
之后就像之前一样使用generator和discriminator就可以了。
这样做的好处是:validate的时候就不需要重新跑一次所有的程序了,只需要把之前的模型 load 出来用就行了
十朵玫瑰代表什么意思?完美的爱情十支玫瑰花代表完美的爱情,十全十美。送10朵玫瑰就表示希望给予对方完美的爱,代表想全心全意爱着对方。
近日,网传一段视频显示,一女子实名举报父亲曾在2021年被多人殴打致死,网传该女孩也被人带走后失联。是否已被找回,对于涉事施害者二刘,去
(原标题:ST摩登:广东佳盟商贸有限公司没有增持)同花顺(300033)金融研究中心12月6日讯,有投资者向ST摩登(002656)提问,你好,广东佳盟商贸有
记者从河南省教育考试院了解到,鉴于目前疫情防控严峻形势,为维护广大考生健康安全,经研究并报教育部有关部门批准,河南省原定于6月11日
尖嘴猴腮是人们对猴子的固有认识,但凡事都有例外,最近,郑州市动物园就有一只猴因长着一张方方正正的脸,被送外号人脸猴,其独特的长相令
中原区民政局提醒:尽量避免扎堆,只要感情好,每一天都是好日子5月20日因谐音我爱你,成为很多新人眼中寓意美好的好日子。为满足准新人们
5月19日是中国旅游日,今年中国旅游日的活动主题为感悟中华文化享受美好旅程。记者今日从省、市文旅部门获悉,为进一步提振行业发展信心,
浊波浩浩东倾,今来古往无终极。在中国所有的大江大河中,无论从地理空间还是文化意义上来说,黄河都是一个独特的存在。如果从高空中俯瞰,
在第46个国际博物馆日当天,国家十三五重点文化工程、位于河南洛阳的隋唐大运河文化博物馆,正式面向公众开放。隋唐大运河始建于公元605年