From 30653181bab5dc2d9276bd0b54306638041d00f4 Mon Sep 17 00:00:00 2001 From: James Betker Date: Wed, 1 Jul 2020 16:30:40 -0600 Subject: [PATCH] Experiment: get rid of post_switch_conv --- codes/models/archs/SwitchedResidualGenerator_arch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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