diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-12-05 20:24:53 +0100 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-12-05 20:24:53 +0100 |
| commit | c18a18b803e34687e541d214c8d67e72e3d0c3cb (patch) | |
| tree | faca7e5d6135cd0a2378b6a2505162d7b4689869 /text_recognizer/networks/efficientnet/mbconv.py | |
| parent | 95d78c2bc2f6c5851cd1124b109077a04b526f10 (diff) | |
Format files with black
Diffstat (limited to 'text_recognizer/networks/efficientnet/mbconv.py')
| -rw-r--r-- | text_recognizer/networks/efficientnet/mbconv.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/text_recognizer/networks/efficientnet/mbconv.py b/text_recognizer/networks/efficientnet/mbconv.py index beb7d57..6324191 100644 --- a/text_recognizer/networks/efficientnet/mbconv.py +++ b/text_recognizer/networks/efficientnet/mbconv.py @@ -158,7 +158,10 @@ 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: |