From 39804692bd4535042e6134ea05ad0e75da82fd13 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sat, 6 Nov 2021 11:47:12 +0100 Subject: Format efficientnet --- text_recognizer/networks/encoders/efficientnet/efficientnet.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'text_recognizer/networks/encoders') diff --git a/text_recognizer/networks/encoders/efficientnet/efficientnet.py b/text_recognizer/networks/encoders/efficientnet/efficientnet.py index 2237acf..7023297 100644 --- a/text_recognizer/networks/encoders/efficientnet/efficientnet.py +++ b/text_recognizer/networks/encoders/efficientnet/efficientnet.py @@ -82,7 +82,9 @@ 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