diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-30 23:02:24 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-30 23:02:24 +0200 |
| commit | 08d73ff01e5e0590e11d5d44a3c85a16bca76ce5 (patch) | |
| tree | 07c985354e11a057681dcc02c79a9225153440aa /text_recognizer/criterions/n_layer_discriminator.py | |
| parent | 1d67f132a400a332eff36555cf5ae3529f436352 (diff) | |
Update GAN loss with sigmoid fn
Diffstat (limited to 'text_recognizer/criterions/n_layer_discriminator.py')
| -rw-r--r-- | text_recognizer/criterions/n_layer_discriminator.py | 1 |
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, |