diff options
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 |