From 3fa2733b697bfe1c1f8a6031ae7f1f66343d5a5b Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sat, 11 Jun 2022 23:10:23 +0200 Subject: Update ff import --- text_recognizer/networks/transformer/decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/text_recognizer/networks/transformer/decoder.py b/text_recognizer/networks/transformer/decoder.py index db5c766..8ea021b 100644 --- a/text_recognizer/networks/transformer/decoder.py +++ b/text_recognizer/networks/transformer/decoder.py @@ -5,7 +5,7 @@ from typing import Optional, Type from torch import nn, Tensor from text_recognizer.networks.transformer.attention import Attention -from text_recognizer.networks.transformer.mlp import FeedForward +from text_recognizer.networks.transformer.ff import FeedForward class DecoderBlock(nn.Module): -- cgit v1.2.3-70-g09d2