Update main.py (pytorch#1095)

val data should not shuffle
mpc001 · Nov 29, 2022 · 387ce7b · 387ce7b
1 parent 1b36393
commit 387ce7b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vae/main.py b/vae/main.py
@@ -41,7 +41,7 @@
     batch_size=args.batch_size, shuffle=True, **kwargs)
 test_loader = torch.utils.data.DataLoader(
     datasets.MNIST('../data', train=False, transform=transforms.ToTensor()),
-    batch_size=args.batch_size, shuffle=True, **kwargs)
+    batch_size=args.batch_size, shuffle=False, **kwargs)
 
 
 class VAE(nn.Module):