summaryrefslogtreecommitdiff
path: root/text_recognizer/networks/encoders/efficientnet/mbconv.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-24 00:02:47 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2021-06-24 00:02:47 +0200
commit1d7f674236d0622addc243d15c05a1dd30ca8121 (patch)
tree53d57dcfb4f2bcc8fef010012db08b7bde2a1559 /text_recognizer/networks/encoders/efficientnet/mbconv.py
parent038195369b3909feeeceb006d52f3af11e3081df (diff)
Still working on efficientnet
Diffstat (limited to 'text_recognizer/networks/encoders/efficientnet/mbconv.py')
-rw-r--r--text_recognizer/networks/encoders/efficientnet/mbconv.py6
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