mirror of
https://github.com/yunjey/pytorch-tutorial.git
synced 2025-07-27 20:13:33 +08:00
For truncated BPTT, use built-in detach
For GANs, detach variables of fake images when training discriminator.
This commit is contained in:
@ -102,7 +102,7 @@ for epoch in range(50):
|
||||
|
||||
noise = Variable(torch.randn(images.size(0), 128))
|
||||
fake_images = generator(noise)
|
||||
outputs = discriminator(fake_images)
|
||||
outputs = discriminator(fake_images.detch())
|
||||
fake_loss = criterion(outputs, fake_labels)
|
||||
fake_score = outputs
|
||||
|
||||
@ -131,4 +131,4 @@ for epoch in range(50):
|
||||
|
||||
# Save the Models
|
||||
torch.save(generator.state_dict(), './generator.pkl')
|
||||
torch.save(discriminator.state_dict(), './discriminator.pkl')
|
||||
torch.save(discriminator.state_dict(), './discriminator.pkl')
|
||||
|
Reference in New Issue
Block a user