From 5a5228072ffe015e676ba696dd022145a9f44222 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 19 Sep 2021 21:03:03 +0200 Subject: Remove normalization in a hope to fix instabilities --- text_recognizer/criterions/n_layer_discriminator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'text_recognizer/criterions') 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 -- cgit v1.2.3-70-g09d2