GAN生成对抗网络

一、代码
import torch

import torch.nn as nn

import numpy as np

import matplotlib.pyplot as plt

import os

os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"

# torch.manual_seed(1) # reproducible

# np.random.seed(1)

 

# 超参数

BATCH_SIZE = 64

LR_G = 0.0001 # 生成器的学习率

LR_D = 0.0001 # 判别器的学习率

N_IDEAS = 5 #生成器的启发因子（就是生成器这个神经网络的初始输入层的节点数）

 

ART_COMPONENTS = 15 # 它可能是G可以在画布上绘制的总点

 

PAINT_POINTS = np.vstack([np.linspace(-1, 1, ART_COMPONENTS) for _ in range(BATCH_SIZE)])

# 展示我们美丽的绘画系列

# plt.plot(PAINT_POINTS[0], 2 * np.power(PAINT_POINTS[0], 2) + 1, c='#74BCFF', lw=3, label='upper bound')

# plt.plot(PAINT_POINTS[0], 1 * np.power(PAINT_POINTS[0], 2) + 0, c='#FF9359', lw=3, label='lower bound')

# plt.legend(loc=