diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-17 22:40:54 +0100 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-11-17 22:40:54 +0100 |
commit | c28126f92672c89ebdb9fc150bc16667f70f5515 (patch) | |
tree | 27e2771d066fef6303f5b15be102c5bd057e817c /text_recognizer/networks/encoders/efficientnet/efficientnet.py | |
parent | 001b34f9a295fa8e828a30b1f820c4ef6eef0d32 (diff) |
Format efficientnet
Diffstat (limited to 'text_recognizer/networks/encoders/efficientnet/efficientnet.py')
-rw-r--r-- | text_recognizer/networks/encoders/efficientnet/efficientnet.py | 4 |
1 files changed, 1 insertions, 3 deletions
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 |