summaryrefslogtreecommitdiff
path: root/text_recognizer/criterions
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-19 21:03:03 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-19 21:03:03 +0200
commit5a5228072ffe015e676ba696dd022145a9f44222 (patch)
tree9802a5248547e136abfd49303de6764fb7bd3c38 /text_recognizer/criterions
parentd8899b2c97046e49807ba2a440ee2dda6e2db335 (diff)
Remove normalization in a hope to fix instabilities
Diffstat (limited to 'text_recognizer/criterions')
-rw-r--r--text_recognizer/criterions/n_layer_discriminator.py2
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