summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-18 17:44:09 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-09-18 17:44:09 +0200
commit71ffa32966c7cc60802244c5e16998da5c5c3ce4 (patch)
treef966a3e6c872085200dc955f460dfc3e3a5d6ea1 /text_recognizer
parent1a193f09bd9199609c30a005dbd28f587dce2606 (diff)
Remove TODO in conv transformer
Diffstat (limited to 'text_recognizer')
-rw-r--r--text_recognizer/networks/conv_transformer.py1
1 files changed, 0 insertions, 1 deletions
diff --git a/text_recognizer/networks/conv_transformer.py b/text_recognizer/networks/conv_transformer.py
index b1a101e..ba43b31 100644
--- a/text_recognizer/networks/conv_transformer.py
+++ b/text_recognizer/networks/conv_transformer.py
@@ -74,7 +74,6 @@ class ConvTransformer(nn.Module):
self.token_embedding.weight.data.uniform_(-bound, bound)
self.head.bias.data.zero_()
self.head.weight.data.uniform_(-bound, bound)
- # TODO: Initalize encoder?
def encode(self, x: Tensor) -> Tensor:
"""Encodes an image into a latent feature vector.