summaryrefslogtreecommitdiff
path: root/text_recognizer/networks
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/networks')
-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.