diff --git a/GAN/vanilla_gan/gan_pytorch.py b/GAN/vanilla_gan/gan_pytorch.py
index 0471069..78d8b27 100644
--- a/GAN/vanilla_gan/gan_pytorch.py
+++ b/GAN/vanilla_gan/gan_pytorch.py
@@ -14,7 +14,6 @@
 mb_size = 64
 Z_dim = 100
 X_dim = mnist.train.images.shape[1]
-y_dim = mnist.train.labels.shape[1]
 h_dim = 128
 c = 0
 lr = 1e-3