diff options
Diffstat (limited to 'text_recognizer/networks')
| -rw-r--r-- | text_recognizer/networks/conv_transformer.py | 1 |
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. |