diff --git a/codes/models/archs/SwitchedResidualGenerator_arch.py b/codes/models/archs/SwitchedResidualGenerator_arch.py
index c435c34a..b2dc8faf 100644
--- a/codes/models/archs/SwitchedResidualGenerator_arch.py
+++ b/codes/models/archs/SwitchedResidualGenerator_arch.py
@@ -212,7 +212,7 @@ class ConfigurableSwitchComputer(nn.Module):
 
         outputs, attention = self.switch(xformed, m, True)
         outputs = identity + outputs
-        outputs = identity + self.post_switch_conv(outputs)
+        #outputs = identity + self.post_switch_conv(outputs)
         outputs = outputs * self.scale + self.bias
         if output_attention_weights:
             return outputs, attention