Fix SSGR

2020-09-20 19:07:15 -06:00 · 2020-09-20 19:07:15 -06:00 · 53a5657850
commit 53a5657850
parent 17c569ea62
1 changed files with 1 additions and 1 deletions
--- a/codes/models/archs/StructuredSwitchedGenerator.py
+++ b/codes/models/archs/StructuredSwitchedGenerator.py
@ -159,7 +159,7 @@ class SSGr1(nn.Module):
        self.init_temperature = init_temperature
        self.final_temperature_step = 10000

-    def forward(self, x, *args):
+    def forward(self, x, embedding):
        noise_stds = []
        # The attention_maps debugger outputs <x>. Save that here.
        self.lr = x.detach().cpu()