diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-08 22:27:43 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-07-08 22:27:43 +0200 |
| commit | 3d279b65f19813357ae395e5f72f1efcbd2829f5 (patch) | |
| tree | 594d5f1a4f4049990eb386e0c1a0fa788273240c | |
| parent | 07186625583f6101eee6babc52d57d5c2cb28b05 (diff) | |
Notebook stuff
| -rw-r--r-- | text_recognizer/networks/cnn_tranformer.py | 5 |
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): |