diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-06 11:47:12 +0100 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-06 11:47:26 +0100 |
commit | 39804692bd4535042e6134ea05ad0e75da82fd13 (patch) | |
tree | dd335c56c20294a1fa7da25cd67033f53fedff30 /text_recognizer/networks/encoders/efficientnet | |
parent | 35e5cefd002913dcb241a9e443c9a81af8a8c814 (diff) |
Format efficientnet
Diffstat (limited to 'text_recognizer/networks/encoders/efficientnet')
-rw-r--r-- | text_recognizer/networks/encoders/efficientnet/efficientnet.py | 4 |
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 |