diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-24 00:02:47 +0200 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2021-06-24 00:02:47 +0200 |
commit | 1d7f674236d0622addc243d15c05a1dd30ca8121 (patch) | |
tree | 53d57dcfb4f2bcc8fef010012db08b7bde2a1559 /text_recognizer/networks/encoders/efficientnet/mbconv.py | |
parent | 038195369b3909feeeceb006d52f3af11e3081df (diff) |
Still working on efficientnet
Diffstat (limited to 'text_recognizer/networks/encoders/efficientnet/mbconv.py')
-rw-r--r-- | text_recognizer/networks/encoders/efficientnet/mbconv.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/text_recognizer/networks/encoders/efficientnet/mbconv.py b/text_recognizer/networks/encoders/efficientnet/mbconv.py index 602aeb7..fbb3f22 100644 --- a/text_recognizer/networks/encoders/efficientnet/mbconv.py +++ b/text_recognizer/networks/encoders/efficientnet/mbconv.py @@ -1,5 +1,5 @@ """Mobile inverted residual block.""" -from typing import Optional, Tuple +from typing import Any, Optional, Tuple import torch from torch import nn, Tensor @@ -20,15 +20,15 @@ class MBConvBlock(nn.Module): bn_momentum: float, bn_eps: float, se_ratio: float, - id_skip: bool, expand_ratio: int, image_size: Optional[Tuple[int, int]], + *args: Any, + **kwargs: Any, ) -> None: super().__init__() self.kernel_size = kernel_size self.bn_momentum = bn_momentum self.bn_eps = bn_eps - self.id_skip = id_skip self.in_channels = self.in_channels self.out_channels = out_channels |