From 8ae1b802bb7d7c63cf758e44269e97a4c0788b65 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 8 Jun 2022 08:40:46 +0200 Subject: Add efficient self attention --- text_recognizer/networks/conformer/__init__.py | 1 + 1 file changed, 1 insertion(+) (limited to 'text_recognizer/networks/conformer/__init__.py') 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 -- cgit v1.2.3-70-g09d2