summaryrefslogtreecommitdiff
path: root/text_recognizer
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-08 22:27:43 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-07-08 22:27:43 +0200
commit3d279b65f19813357ae395e5f72f1efcbd2829f5 (patch)
tree594d5f1a4f4049990eb386e0c1a0fa788273240c /text_recognizer
parent07186625583f6101eee6babc52d57d5c2cb28b05 (diff)
Notebook stuff
Diffstat (limited to 'text_recognizer')
-rw-r--r--text_recognizer/networks/cnn_tranformer.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/text_recognizer/networks/cnn_tranformer.py b/text_recognizer/networks/cnn_tranformer.py
index 38de0ba..ff0ae82 100644
--- a/text_recognizer/networks/cnn_tranformer.py
+++ b/text_recognizer/networks/cnn_tranformer.py
@@ -4,6 +4,11 @@ from typing import Type
import attr
from torch import nn, Tensor
+from text_recognizer.networks.transformer.positional_encodings import (
+ PositionalEncoding,
+ PositionalEncoding2D,
+)
+
@attr.s
class CnnTransformer(nn.Module):