diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-19 21:03:03 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-19 21:03:03 +0200 |
| commit | 5a5228072ffe015e676ba696dd022145a9f44222 (patch) | |
| tree | 9802a5248547e136abfd49303de6764fb7bd3c38 /text_recognizer/criterions/n_layer_discriminator.py | |
| parent | d8899b2c97046e49807ba2a440ee2dda6e2db335 (diff) | |
Remove normalization in a hope to fix instabilities
Diffstat (limited to 'text_recognizer/criterions/n_layer_discriminator.py')
| -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 |