diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-18 17:44:09 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-09-18 17:44:09 +0200 |
| commit | 71ffa32966c7cc60802244c5e16998da5c5c3ce4 (patch) | |
| tree | f966a3e6c872085200dc955f460dfc3e3a5d6ea1 | |
| parent | 1a193f09bd9199609c30a005dbd28f587dce2606 (diff) | |
Remove TODO in conv transformer
| -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. |