summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/convnext
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2023-01-29 21:16:22 +0100
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2023-01-29 21:16:22 +0100
commit5bbbec1f7021c2451099eb0d1e00a330275d1e16 (patch)
treee98be46558eaef946e1727b9d05cd9b90369d2af /text_recognizer/networks/convnext
parent1a0b6d032c1b7e5cffffed750385854fbf5b239c (diff)
Update import
Diffstat (limited to 'text_recognizer/networks/convnext')
-rw-r--r--text_recognizer/networks/convnext/attention.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/convnext/attention.py b/text_recognizer/networks/convnext/attention.py
index a373c37..1334feb 100644
--- a/text_recognizer/networks/convnext/attention.py
+++ b/text_recognizer/networks/convnext/attention.py
@@ -1,7 +1,7 @@
"""Convolution self attention block."""
import torch.nn.functional as F
-from einops import rearrange, reduce
+from einops import rearrange
from torch import Tensor, einsum, nn
from text_recognizer.networks.convnext.norm import LayerNorm