diff options
Diffstat (limited to 'text_recognizer/networks/transformer/decoder.py')
-rw-r--r-- | text_recognizer/networks/transformer/decoder.py | 2 |
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 |