From 71ffa32966c7cc60802244c5e16998da5c5c3ce4 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sat, 18 Sep 2021 17:44:09 +0200 Subject: Remove TODO in conv transformer --- text_recognizer/networks/conv_transformer.py | 1 - 1 file changed, 1 deletion(-) (limited to 'text_recognizer') 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. -- cgit v1.2.3-70-g09d2