summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/transformer/decoder.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-18 18:11:21 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-18 18:11:53 +0200
commit2cc6aa059139b57057609817913ad515063c2eab (patch)
tree5433f69a5eaf63e064a100bf900783127c7b1ff4 /text_recognizer/networks/transformer/decoder.py
parent88caa5c466225d4752541c352c5777235f8f0c61 (diff)
Format imports
Format imports
Diffstat (limited to 'text_recognizer/networks/transformer/decoder.py')
-rw-r--r--text_recognizer/networks/transformer/decoder.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/text_recognizer/networks/transformer/decoder.py b/text_recognizer/networks/transformer/decoder.py
index 8ea021b..efa1e89 100644
--- a/text_recognizer/networks/transformer/decoder.py
+++ b/text_recognizer/networks/transformer/decoder.py
@@ -2,7 +2,7 @@
from copy import deepcopy
from typing import Optional, Type
-from torch import nn, Tensor
+from torch import Tensor, nn
from text_recognizer.networks.transformer.attention import Attention
from text_recognizer.networks.transformer.ff import FeedForward