mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Experiment: get rid of post_switch_conv

Browse Source
This commit is contained in:
James Betker 2020-07-01 16:30:40 -06:00
parent 17191de836
commit 30653181ba
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
codes/models/archs/SwitchedResidualGenerator_arch.py
View File

@ -212,7 +212,7 @@ class ConfigurableSwitchComputer(nn.Module):
outputs, attention = self.switch(xformed, m, True) outputs, attention = self.switch(xformed, m, True)
outputs = identity + outputs outputs = identity + outputs
outputs = identity + self.post_switch_conv(outputs) #outputs = identity + self.post_switch_conv(outputs)
outputs = outputs * self.scale + self.bias outputs = outputs * self.scale + self.bias
if output_attention_weights: if output_attention_weights:
return outputs, attention return outputs, attention