From 5a5228072ffe015e676ba696dd022145a9f44222 Mon Sep 17 00:00:00 2001
From: Gustaf Rydholm <gustaf.rydholm@gmail.com>
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')
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