From 0d59b196bd434e11146da60e1ed24fc92aafad03 Mon Sep 17 00:00:00 2001 From: kunkunlin Date: Wed, 25 Dec 2024 23:01:41 +0800 Subject: [PATCH] [F] Fix SeparableConv2dBlock module list error --- chameleon/base/blocks/conv_block.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/chameleon/base/blocks/conv_block.py b/chameleon/base/blocks/conv_block.py index 6afb808..372976a 100644 --- a/chameleon/base/blocks/conv_block.py +++ b/chameleon/base/blocks/conv_block.py @@ -67,6 +67,10 @@ def __init__( groups=in_channels, bias=False, ) + if inner_norm is not None: + self.inner_norm = COMPONENTS.build(inner_norm) if isinstance(inner_norm, dict) else inner_norm + if inner_act is not None: + self.inner_act = COMPONENTS.build(inner_act) if isinstance(inner_act, dict) else inner_act self.pw_conv = nn.Conv2d( in_channels, out_channels, @@ -75,10 +79,6 @@ def __init__( padding=0, bias=bias, ) - if inner_norm is not None: - self.inner_norm = COMPONENTS.build(inner_norm) if isinstance(inner_norm, dict) else inner_norm - if inner_act is not None: - self.inner_act = COMPONENTS.build(inner_act) if isinstance(inner_act, dict) else inner_act if norm is not None: self.norm = COMPONENTS.build(norm) if isinstance(norm, dict) else norm if act is not None: