summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-11 15:44:51 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-11 15:44:51 +0200
commit7eef87c2c72412714b8ede9bba49196791f443b5 (patch)
tree10cea28be9f821d5db1ab4e29bd7bfb5863b7b13 /text_recognizer
parent9c829df67f0a874b2803769dc8ff3681a3c095b1 (diff)
Disable group norm in vqvae en/decoder
Diffstat (limited to 'text_recognizer')
-rw-r--r--text_recognizer/networks/vqvae/decoder.py2
-rw-r--r--text_recognizer/networks/vqvae/encoder.py2
2 files changed, 2 insertions, 2 deletions
diff --git a/text_recognizer/networks/vqvae/decoder.py b/text_recognizer/networks/vqvae/decoder.py
index fcbed57..1194573 100644
--- a/text_recognizer/networks/vqvae/decoder.py
+++ b/text_recognizer/networks/vqvae/decoder.py
@@ -37,7 +37,7 @@ class Decoder(nn.Module):
in_channels=in_channels,
out_channels=in_channels,
dropout_rate=self.dropout_rate,
- use_norm=True,
+ use_norm=False,
),
]
diff --git a/text_recognizer/networks/vqvae/encoder.py b/text_recognizer/networks/vqvae/encoder.py
index 4a5c976..06e6c18 100644
--- a/text_recognizer/networks/vqvae/encoder.py
+++ b/text_recognizer/networks/vqvae/encoder.py
@@ -63,7 +63,7 @@ class Encoder(nn.Module):
in_channels=self.hidden_dim * self.channels_multipliers[-1],
out_channels=self.hidden_dim * self.channels_multipliers[-1],
dropout_rate=self.dropout_rate,
- use_norm=True,
+ use_norm=False,
)
]