summaryrefslogtreecommitdiff
path: root/text_recognizer/network/transformer/encoder.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2023-09-03 01:14:16 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2023-09-03 01:14:16 +0200
commitd020059f2f71fe7c25765dde9d535195c09ece01 (patch)
treefbef1ff2b5d4475e6d3117575c61bde77802d07d /text_recognizer/network/transformer/encoder.py
parentc66ca7420560e9ab0e8893f6c7d6a496f128c79a (diff)
Update imports
Diffstat (limited to 'text_recognizer/network/transformer/encoder.py')
-rw-r--r--text_recognizer/network/transformer/encoder.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/text_recognizer/network/transformer/encoder.py b/text_recognizer/network/transformer/encoder.py
index ea4b0b3..328a40c 100644
--- a/text_recognizer/network/transformer/encoder.py
+++ b/text_recognizer/network/transformer/encoder.py
@@ -1,8 +1,8 @@
"""Transformer encoder module."""
from torch import Tensor, nn
-from text_recognizer.network.transformer.attention import Attention
-from text_recognizer.network.transformer.ff import FeedForward
+from .attention import Attention
+from .ff import FeedForward
class Encoder(nn.Module):