From 08d73ff01e5e0590e11d5d44a3c85a16bca76ce5 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
Date: Thu, 30 Sep 2021 23:02:24 +0200
Subject: Update GAN loss with sigmoid fn

---
 text_recognizer/criterions/n_layer_discriminator.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/text_recognizer/criterions/n_layer_discriminator.py b/text_recognizer/criterions/n_layer_discriminator.py
index 7df724d..a9f47f9 100644
--- a/text_recognizer/criterions/n_layer_discriminator.py
+++ b/text_recognizer/criterions/n_layer_discriminator.py
@@ -19,6 +19,7 @@ class NLayerDiscriminator(nn.Module):
     def _build_discriminator(self) -> nn.Sequential:
         """Builds discriminator."""
         discriminator = [
+            nn.Sigmoid(),
             nn.Conv2d(
                 in_channels=self.in_channels,
                 out_channels=self.num_channels,
-- 
cgit v1.2.3-70-g09d2