summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/conformer/__init__.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-08 08:40:46 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-08 08:40:46 +0200
commit8ae1b802bb7d7c63cf758e44269e97a4c0788b65 (patch)
treefdabe515ace3e223c45dfbe72e3a4498abefbce3 /text_recognizer/networks/conformer/__init__.py
parentd194ab9fd3eb64d715c83066a7a690c6c8834dde (diff)
Add efficient self attention
Diffstat (limited to 'text_recognizer/networks/conformer/__init__.py')
-rw-r--r--text_recognizer/networks/conformer/__init__.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/text_recognizer/networks/conformer/__init__.py b/text_recognizer/networks/conformer/__init__.py
index 1886f85..8951481 100644
--- a/text_recognizer/networks/conformer/__init__.py
+++ b/text_recognizer/networks/conformer/__init__.py
@@ -4,3 +4,4 @@ from text_recognizer.networks.conformer.glu import GLU
from text_recognizer.networks.conformer.conformer import Conformer
from text_recognizer.networks.conformer.conv import ConformerConv
from text_recognizer.networks.conformer.subsampler import Subsampler
+from text_recognizer.networks.conformer.attention import Attention