summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2024-04-15 21:49:02 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2024-04-15 21:49:02 +0200
commitc7e5354ffa43eccfc4e411375ce2f531af7bbcff (patch)
tree9147289d339e5ba3960d598695e6cd226bff4bed
parente643e0c61ab33ce1bb8cfdebc92fc0670c82afda (diff)
Format
-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):