summaryrefslogtreecommitdiff
path: root/text_recognizer/criterions
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-30 23:02:24 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-30 23:02:24 +0200
commit08d73ff01e5e0590e11d5d44a3c85a16bca76ce5 (patch)
tree07c985354e11a057681dcc02c79a9225153440aa /text_recognizer/criterions
parent1d67f132a400a332eff36555cf5ae3529f436352 (diff)
Update GAN loss with sigmoid fn
Diffstat (limited to 'text_recognizer/criterions')
-rw-r--r--text_recognizer/criterions/n_layer_discriminator.py1
1 files changed, 1 insertions, 0 deletions
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,