目录
-
- 一、生成器
- 二、判别器
- 三、迭代训练过程
- 四、结果演示
- 五、完整代码
LSGAN的架构与DCGAN基本类似,只是改了损失函数,将损失函数从BCEloss改为MSEloss。
一、生成器生成器类与DCGAN基本相同,代码如下:
class Generator(nn.Module):
def __init__(self):
super(