Skip to content

Commit 46826b0

Browse files
committed
fix normalization
1 parent 44bba3e commit 46826b0

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

SP21/GAN/vanilla_gan.ipynb

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -101,9 +101,7 @@
101101
"img_shape = (28, 28)\n",
102102
"batch_size = 64\n",
103103
"\n",
104-
"transform = transforms.Compose([\n",
105-
" transforms.ToTensor(),\n",
106-
" transforms.Normalize(mean=(0.5), std=(0.5))])\n",
104+
"transform = transforms.Compose([transforms.ToTensor()])\n",
107105
"train_dataset = torchvision.datasets.MNIST(root=\"./data\", train = True, download=True, transform=transform)\n",
108106
"train_dataloader = torch.utils.data.DataLoader(train_dataset, batch_size=batch_size, shuffle=True)"
109107
]

0 commit comments

Comments
 (0)