summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/conformer
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-11 23:10:01 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-11 23:10:01 +0200
commit8c69c3d457ed3279c59a8c25b08a49f39562db51 (patch)
tree4ab54e7957663dad808cbabbef0ada48dec2d103 /text_recognizer/networks/conformer
parent6df941bdf5cad80db38d851dcb23a08a9dc55617 (diff)
Format
Diffstat (limited to 'text_recognizer/networks/conformer')
-rw-r--r--text_recognizer/networks/conformer/subsampler.py4
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):