summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--text_recognizer/network/mammut.py6
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):