[Bug Fixed] Correct the number of output channels of LFF in RDB (#1292)

Correct the out_channels of LFF
open-mmlab · Oct 13, 2022 · 14f1ffc · 14f1ffc
1 parent 27f14e4
commit 14f1ffc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmedit/models/backbones/sr_backbones/rdn.py b/mmedit/models/backbones/sr_backbones/rdn.py
@@ -51,7 +51,7 @@ def __init__(self, in_channels, channel_growth, num_layers):
         # local feature fusion
         self.lff = nn.Conv2d(
             in_channels + channel_growth * num_layers,
-            channel_growth,
+            in_channels,
             kernel_size=1)
 
     def forward(self, x):