mirror of
https://github.com/yunjey/pytorch-tutorial.git
synced 2025-07-28 20:53:44 +08:00
add denormalization function
This commit is contained in:
@ -11,8 +11,11 @@ transform = transforms.Compose([
|
||||
transforms.ToTensor(),
|
||||
transforms.Normalize(mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5))])
|
||||
|
||||
def denorm(x):
|
||||
return (x + 1) / 2
|
||||
|
||||
# MNIST Dataset
|
||||
train_dataset = dsets.MNIST(root='../data/',
|
||||
train_dataset = dsets.MNIST(root='./data/',
|
||||
train=True,
|
||||
transform=transform,
|
||||
download=True)
|
||||
@ -102,9 +105,9 @@ for epoch in range(200):
|
||||
|
||||
# Save the sampled images
|
||||
fake_images = fake_images.view(fake_images.size(0), 1, 28, 28)
|
||||
torchvision.utils.save_image(fake_images.data,
|
||||
torchvision.utils.save_image(denorm(fake_images.data),
|
||||
'./data/fake_samples_%d.png' %(epoch+1))
|
||||
|
||||
# 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