Fix (sdv-dev#160)

JonathanDZiegler · Jul 14, 2021 · 2dbbc18 · 2dbbc18
1 parent 99d403a
commit 2dbbc18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ctgan/synthesizers/ctgan.py b/ctgan/synthesizers/ctgan.py
@@ -360,7 +360,7 @@ def fit(self, train_data, discrete_columns=tuple(), epochs=None):
                         real_cat = torch.cat([real, c2], dim=1)
                     else:
                         real_cat = real
-                        fake_cat = fake
+                        fake_cat = fakeact
 
                     y_fake = discriminator(fake_cat)
                     y_real = discriminator(real_cat)