From 2cc6aa059139b57057609817913ad515063c2eab Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 18 Sep 2022 18:11:21 +0200 Subject: Format imports Format imports --- text_recognizer/networks/transformer/embeddings/axial.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'text_recognizer/networks/transformer/embeddings/axial.py') diff --git a/text_recognizer/networks/transformer/embeddings/axial.py b/text_recognizer/networks/transformer/embeddings/axial.py index 7b84e12..25d8f60 100644 --- a/text_recognizer/networks/transformer/embeddings/axial.py +++ b/text_recognizer/networks/transformer/embeddings/axial.py @@ -3,10 +3,11 @@ Stolen from: https://github.com/lucidrains/axial-attention/blob/eff2c10c2e76c735a70a6b995b571213adffbbb7/axial_attention/axial_attention.py#L100 """ +from functools import reduce +from operator import mul + import torch from torch import nn -from operator import mul -from functools import reduce class AxialPositionalEmbedding(nn.Module): -- cgit v1.2.3-70-g09d2