Another Bug Fixed: Should feed raw logits to CELoss rather than logprobs

corenel · Apr 12, 2022 · 6e59bbd · 6e59bbd
1 parent 13a295a
commit 6e59bbd
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/models/discriminator.py b/models/discriminator.py
@@ -18,7 +18,6 @@ def __init__(self, input_dims, hidden_dims, output_dims):
             nn.Linear(hidden_dims, hidden_dims),
             nn.ReLU(),
             nn.Linear(hidden_dims, output_dims),
-            nn.LogSoftmax()
         )
 
     def forward(self, input):