deviandice/DL-Art-School
1
0
Fork 0
forked from mrq/DL-Art-School
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Stacked switches - return a tuple

Browse Source
This commit is contained in:
James Betker 2020-10-04 21:02:24 -06:00
parent 13f97e1e97
commit c3ef8a4a31
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/StructuredSwitchedGenerator.py
View File

@ -281,7 +281,7 @@ class StackedSwitchGenerator(nn.Module):
x_out = checkpoint(self.final_hr_conv2, x_out) x_out = checkpoint(self.final_hr_conv2, x_out)
self.attentions = [a1, a3, a3] self.attentions = [a1, a3, a3]
return x_out return x_out,
def set_temperature(self, temp): def set_temperature(self, temp):
[sw.set_temperature(temp) for sw in self.switches] [sw.set_temperature(temp) for sw in self.switches]