diff options
Diffstat (limited to 'text_recognizer/networks/conformer')
| -rw-r--r-- | text_recognizer/networks/conformer/subsampler.py | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/text_recognizer/networks/conformer/subsampler.py b/text_recognizer/networks/conformer/subsampler.py index 42a983e..133b53a 100644 --- a/text_recognizer/networks/conformer/subsampler.py +++ b/text_recognizer/networks/conformer/subsampler.py @@ -4,9 +4,7 @@ from typing import Tuple from einops import rearrange from torch import nn, Tensor -from text_recognizer.networks.transformer import ( - AxialPositionalEmbedding, -) +from text_recognizer.networks.transformer import AxialPositionalEmbedding class Subsampler(nn.Module): |