训练循环
for epoch in range(100):
for i, data in enumerate(dataset):
训练判别器
optimizerD.zero_grad()
real_data = data
z = torch.randn(100, 100)
fake_data = generator(z)
real_output = discriminator(real_data)
fake_output = discriminator(fake_data.detach())
lossD = criterion(real_output, torch.ones_like(real_output)) + criterion(fake_output, torch.zeros_like(fake_output))
lossD.backward()
optimizerD.step()