From c28126f92672c89ebdb9fc150bc16667f70f5515 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Wed, 17 Nov 2021 22:40:54 +0100 Subject: Format efficientnet --- text_recognizer/networks/encoders/efficientnet/efficientnet.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) (limited to 'text_recognizer/networks/encoders/efficientnet/efficientnet.py') diff --git a/text_recognizer/networks/encoders/efficientnet/efficientnet.py b/text_recognizer/networks/encoders/efficientnet/efficientnet.py index 8c1ebcf..9454514 100644 --- a/text_recognizer/networks/encoders/efficientnet/efficientnet.py +++ b/text_recognizer/networks/encoders/efficientnet/efficientnet.py @@ -82,9 +82,7 @@ class EfficientNet(nn.Module): for _ in range(num_repeats): self._blocks.append( MBConvBlock( - **args, - bn_momentum=self.bn_momentum, - bn_eps=self.bn_eps, + **args, bn_momentum=self.bn_momentum, bn_eps=self.bn_eps, ) ) args.in_channels = args.out_channels -- cgit v1.2.3-70-g09d2