目录
一、生成器
-
- 一、生成器
- 二、判别器
- 三、迭代训练过程
- 四、结果演示
- 五、完整代码
生成器的目标是输入一个固定维度的随机噪声,生成手写数字图像,因此它的输入是一个长度为N=100的一维的向量,输出一个28,28,1维的图像。 输入一个100维的0~1之间的高斯分布,然后通过第一层线性变换将其映射到128维,然后通过LeakyReLU激活函数,接着进行一个线性变换将其变到256维,再经过一个LeakyReLU激活函数,然后经过线性变换将其变成512维,再经过一个LeakyReLU激活函数,然后经过线性变换将其变成1024维,最后经过线性变换将其变成784维(即28281维),最后经过Tanh激活函数是希望生成的假的图片数据分布能够在-1~1之间。 生成器类代码:
class Generator(nn.Module)