summaryrefslogtreecommitdiff
path: root/text_recognizer/network/transformer/embedding
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/network/transformer/embedding')
-rw-r--r--text_recognizer/network/transformer/embedding/absolute.py3
-rw-r--r--text_recognizer/network/transformer/embedding/token.py2
2 files changed, 3 insertions, 2 deletions
diff --git a/text_recognizer/network/transformer/embedding/absolute.py b/text_recognizer/network/transformer/embedding/absolute.py
index 08b2c2a..db34157 100644
--- a/text_recognizer/network/transformer/embedding/absolute.py
+++ b/text_recognizer/network/transformer/embedding/absolute.py
@@ -2,7 +2,8 @@ from typing import Optional
import torch
from torch import nn, Tensor
-from text_recognizer.network.transformer.embedding.l2_norm import l2_norm
+
+from .l2_norm import l2_norm
class AbsolutePositionalEmbedding(nn.Module):
diff --git a/text_recognizer/network/transformer/embedding/token.py b/text_recognizer/network/transformer/embedding/token.py
index 1df2fd6..838f514 100644
--- a/text_recognizer/network/transformer/embedding/token.py
+++ b/text_recognizer/network/transformer/embedding/token.py
@@ -1,6 +1,6 @@
from torch import nn, Tensor
-from text_recognizer.network.transformer.embedding.l2_norm import l2_norm
+from .l2_norm import l2_norm
class TokenEmbedding(nn.Module):