summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-06 11:47:12 +0100
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-11-06 11:47:26 +0100
commit39804692bd4535042e6134ea05ad0e75da82fd13 (patch)
treedd335c56c20294a1fa7da25cd67033f53fedff30
parent35e5cefd002913dcb241a9e443c9a81af8a8c814 (diff)
Format efficientnet
-rw-r--r--text_recognizer/networks/encoders/efficientnet/efficientnet.py4
1 files changed, 3 insertions, 1 deletions
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