diff options
Diffstat (limited to 'text_recognizer/criterions')
| -rw-r--r-- | text_recognizer/criterions/n_layer_discriminator.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/criterions/n_layer_discriminator.py b/text_recognizer/criterions/n_layer_discriminator.py index e5f8449..7df724d 100644 --- a/text_recognizer/criterions/n_layer_discriminator.py +++ b/text_recognizer/criterions/n_layer_discriminator.py @@ -38,7 +38,7 @@ class NLayerDiscriminator(nn.Module): stride=2, padding=1, ), - Normalize(num_channels=in_channels * n), + # Normalize(num_channels=in_channels * n), nn.Mish(inplace=True), ] in_channels *= n |