diff --git a/src/Restorer/Cond_NAF.py b/src/Restorer/Cond_NAF.py index 1deaa46..432e0bf 100644 --- a/src/Restorer/Cond_NAF.py +++ b/src/Restorer/Cond_NAF.py @@ -215,7 +215,7 @@ def __init__(self, c, DW_Expand=2, FFN_Expand=2, drop_out_rate=0.0, cond_chans=0 # Simplified Channel Attention self.sca = ConditionedChannelAttention(dw_channel // 2, cond_chans) - self.sca_out = ConditionedChannelAttention(c, cond_chans) + # self.sca_out = ConditionedChannelAttention(c, cond_chans) # SimpleGate self.sg = SimpleGate() @@ -281,7 +281,7 @@ def forward(self, input): x = self.dropout2(x) - x = self.sca_out(x, cond) * x + # x = self.sca_out(x, cond) * x return (y + x * self.gamma, cond)