From ad63099ce19fcfbee577c282473c1a4b2f12450b Mon Sep 17 00:00:00 2001 From: Thien Tran Date: Mon, 24 Jul 2023 10:09:22 +0800 Subject: [PATCH] add typing --- vision_toolbox/backbones/mobile_vit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vision_toolbox/backbones/mobile_vit.py b/vision_toolbox/backbones/mobile_vit.py index 58ffc24..f050c9f 100644 --- a/vision_toolbox/backbones/mobile_vit.py +++ b/vision_toolbox/backbones/mobile_vit.py @@ -92,7 +92,7 @@ class MobileViTStageConfig(NamedTuple): class MobileViT(BaseBackbone): - def __init__(self, stem_channels: int, stage_configs: list[MobileViTStageConfig], out_channels: int): + def __init__(self, stem_channels: int, stage_configs: list[MobileViTStageConfig], out_channels: int) -> None: super().__init__() self.stem = nn.Sequential( ConvNormAct(3, 16, 3, 2, act="swish"),