From 53a56578502053270e45701d551fa85cb8d228b7 Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Sun, 20 Sep 2020 19:07:15 -0600
Subject: [PATCH] Fix SSGR

---
 codes/models/archs/StructuredSwitchedGenerator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/codes/models/archs/StructuredSwitchedGenerator.py b/codes/models/archs/StructuredSwitchedGenerator.py
index c68b8619..714df7a6 100644
--- 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()