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/mbconv.py | |
| parent | 001b34f9a295fa8e828a30b1f820c4ef6eef0d32 (diff) | |
Format efficientnet
Diffstat (limited to 'text_recognizer/networks/encoders/efficientnet/mbconv.py')
| -rw-r--r-- | text_recognizer/networks/encoders/efficientnet/mbconv.py | 5 |
1 files changed, 1 insertions, 4 deletions
diff --git a/text_recognizer/networks/encoders/efficientnet/mbconv.py b/text_recognizer/networks/encoders/efficientnet/mbconv.py index 96626fc..4b051eb 100644 --- a/text_recognizer/networks/encoders/efficientnet/mbconv.py +++ b/text_recognizer/networks/encoders/efficientnet/mbconv.py @@ -158,10 +158,7 @@ class MBConvBlock(nn.Module): def _configure_padding(self) -> Tuple[int, int, int, int]: """Set padding for convolutional layers.""" if self.stride == (2, 2): - return ( - (self.kernel_size - 1) // 2 - 1, - (self.kernel_size - 1) // 2, - ) * 2 + return ((self.kernel_size - 1) // 2 - 1, (self.kernel_size - 1) // 2,) * 2 return ((self.kernel_size - 1) // 2,) * 4 def __attrs_post_init__(self) -> None: |