diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-06 23:19:35 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-06 23:19:35 +0200 |
| commit | 01d6e5fc066969283df99c759609df441151e9c5 (patch) | |
| tree | ecd1459e142356d0c7f50a61307b760aca813248 /text_recognizer/networks/transformer/positional_encodings | |
| parent | f4688482b4898c0b342d6ae59839dc27fbf856c6 (diff) | |
Working on fixing decoder transformer
Diffstat (limited to 'text_recognizer/networks/transformer/positional_encodings')
| -rw-r--r-- | text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py b/text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py index 9466f6e..7140537 100644 --- a/text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py +++ b/text_recognizer/networks/transformer/positional_encodings/absolute_embedding.py @@ -1,4 +1,5 @@ """Absolute positional embedding.""" +import torch from torch import nn, Tensor |