diff options
Diffstat (limited to 'text_recognizer/network/mammut.py')
| -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): |