diff --git a/mmedit/models/backbones/encoder_decoders/decoders/fba_decoder.py b/mmedit/models/backbones/encoder_decoders/decoders/fba_decoder.py index fe01e35c4a..0d55c337a6 100644 --- a/mmedit/models/backbones/encoder_decoders/decoders/fba_decoder.py +++ b/mmedit/models/backbones/encoder_decoders/decoders/fba_decoder.py @@ -110,31 +110,26 @@ def __init__(self, self.unpool = nn.MaxUnpool2d(2, stride=2) - self.conv_up4 = nn.Sequential(*(list( - ConvModule( - 64 + 3 + 3 + 2, - 32, - padding=1, - kernel_size=3, - bias=True, - conv_cfg=self.conv_cfg, - act_cfg=self.act_cfg).children()) + list( - ConvModule( - 32, - 16, - padding=1, - kernel_size=3, - bias=True, - conv_cfg=self.conv_cfg, - act_cfg=self.act_cfg).children()) + list( - ConvModule( - 16, - 7, - padding=0, - kernel_size=1, - bias=True, - conv_cfg=self.conv_cfg, - act_cfg=None).children()))) + self.conv_up4 = nn.Sequential( + *(list( + ConvModule( + 64 + 3 + 3 + 2, + 32, + padding=1, + kernel_size=3, + bias=True, + act_cfg=self.act_cfg).children()) + list( + ConvModule( + 32, + 16, + padding=1, + kernel_size=3, + bias=True, + act_cfg=self.act_cfg).children()) + + list( + ConvModule( + 16, 7, padding=0, kernel_size=1, bias=True, + act_cfg=None).children()))) def init_weights(self, pretrained=None): """Init weights for the model.