diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-01-29 21:16:22 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2023-01-29 21:16:22 +0100 |
| commit | 5bbbec1f7021c2451099eb0d1e00a330275d1e16 (patch) | |
| tree | e98be46558eaef946e1727b9d05cd9b90369d2af /text_recognizer/networks/convnext | |
| parent | 1a0b6d032c1b7e5cffffed750385854fbf5b239c (diff) | |
Update import
Diffstat (limited to 'text_recognizer/networks/convnext')
| -rw-r--r-- | text_recognizer/networks/convnext/attention.py | 2 |
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 |