From a2cb31e9f02609842ad3ff40fc356ddabc789025 Mon Sep 17 00:00:00 2001
From: sanctuary <836790896@qq.com>
Date: Wed, 14 Sep 2022 20:37:22 +0800
Subject: [PATCH] modiry acgan

---
 .gitignore                     | 2 +-
 implementations/acgan/acgan.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

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)