diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-01-29 15:56:56 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-01-29 15:56:56 +0100 |
| commit | 262159039c221839792acbdd7a8a6c7b80fb0dac (patch) | |
| tree | fb07ddc8075eee64619faeaa70e7a59dcac898fc /text_recognizer | |
| parent | eae5ca3e561eff0a8adfe4f27f13ccc49691e468 (diff) | |
chore(decoder): add new import path for decoder
Diffstat (limited to 'text_recognizer')
| -rw-r--r-- | text_recognizer/networks/conv_transformer.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/conv_transformer.py b/text_recognizer/networks/conv_transformer.py index 77e4984..a068ea3 100644 --- a/text_recognizer/networks/conv_transformer.py +++ b/text_recognizer/networks/conv_transformer.py @@ -7,10 +7,10 @@ from torch import nn, Tensor from text_recognizer.networks.base import BaseTransformer from text_recognizer.networks.transformer.axial_attention.encoder import AxialEncoder +from text_recognizer.networks.transformer.decoder import Decoder from text_recognizer.networks.transformer.embeddings.axial import ( AxialPositionalEmbedding, ) -from text_recognizer.networks.transformer.layers import Decoder class ConvTransformer(BaseTransformer): |