diff --git a/.gitignore b/.gitignore
index 9fc40d1d..a4f8c195 100644
--- a/.gitignore
+++ b/.gitignore
@@ -8,5 +8,5 @@ data/*/
 implementations/*/data
 implementations/*/images
 implementations/*/saved_models
-
+.history
 __pycache__
diff --git a/implementations/acgan/acgan.py b/implementations/acgan/acgan.py
index d7fa2486..47861537 100644
--- a/implementations/acgan/acgan.py
+++ b/implementations/acgan/acgan.py
@@ -97,7 +97,7 @@ def discriminator_block(in_filters, out_filters, bn=True):
 
         # Output layers
         self.adv_layer = nn.Sequential(nn.Linear(128 * ds_size ** 2, 1), nn.Sigmoid())
-        self.aux_layer = nn.Sequential(nn.Linear(128 * ds_size ** 2, opt.n_classes), nn.Softmax())
+        self.aux_layer = nn.Sequential(nn.Linear(128 * ds_size ** 2, opt.n_classes))
 
     def forward(self, img):
         out = self.conv_blocks(img)