summaryrefslogtreecommitdiff
path: root/text_recognizer/model
diff options
context:
space:
mode:
Diffstat (limited to 'text_recognizer/model')
-rw-r--r--text_recognizer/model/transformer.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/model/transformer.py b/text_recognizer/model/transformer.py
index 598d995..5842bdb 100644
--- a/text_recognizer/model/transformer.py
+++ b/text_recognizer/model/transformer.py
@@ -6,7 +6,7 @@ from omegaconf import DictConfig
from torch import nn, Tensor
from torchmetrics import CharErrorRate, WordErrorRate
-from .greedy_decoder import GreedyDecoder
+from text_recognizer.decoder.greedy_decoder import GreedyDecoder
from text_recognizer.data.tokenizer import Tokenizer
from .base import LitBase