Experiment: get rid of post_switch_conv

2020-07-01 16:30:40 -06:00 · 2020-07-01 16:30:40 -06:00 · 30653181ba
commit 30653181ba
parent 17191de836
1 changed files with 1 additions and 1 deletions
--- 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