diff options
Diffstat (limited to 'text_recognizer/criterions')
| -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, |