diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-15 21:49:02 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2024-04-15 21:49:02 +0200 |
| commit | c7e5354ffa43eccfc4e411375ce2f531af7bbcff (patch) | |
| tree | 9147289d339e5ba3960d598695e6cd226bff4bed /text_recognizer | |
| parent | e643e0c61ab33ce1bb8cfdebc92fc0670c82afda (diff) | |
Format
Diffstat (limited to 'text_recognizer')
| -rw-r--r-- | text_recognizer/network/mammut.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/text_recognizer/network/mammut.py b/text_recognizer/network/mammut.py index 761e5f9..8754420 100644 --- a/text_recognizer/network/mammut.py +++ b/text_recognizer/network/mammut.py @@ -1,12 +1,12 @@ from typing import Type -from einops import repeat import torch -from torch import nn, Tensor import torch.nn.functional as F +from einops import repeat +from torch import Tensor, nn -from text_recognizer.network.transformer.decoder import Decoder from text_recognizer.network.transformer.attention import Attention +from text_recognizer.network.transformer.decoder import Decoder class ToLatent(nn.Module): |