Stacked switches - return a tuple

2020-10-04 21:02:24 -06:00 · 2020-10-04 21:02:24 -06:00 · c3ef8a4a31
commit c3ef8a4a31
parent 13f97e1e97
1 changed files with 1 additions and 1 deletions
--- a/codes/models/archs/StructuredSwitchedGenerator.py
+++ b/codes/models/archs/StructuredSwitchedGenerator.py
@ -281,7 +281,7 @@ class StackedSwitchGenerator(nn.Module):
        x_out = checkpoint(self.final_hr_conv2, x_out)

        self.attentions = [a1, a3, a3]
-        return x_out
+        return x_out,

    def set_temperature(self, temp):
        [sw.set_temperature(temp) for sw in self.switches]