From 262159039c221839792acbdd7a8a6c7b80fb0dac Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sat, 29 Jan 2022 15:56:56 +0100 Subject: chore(decoder): add new import path for decoder --- text_recognizer/networks/conv_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'text_recognizer') 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): -- cgit v1.2.3-70-g09d2