Skip to content

Commit b5900f2

Browse files
authored
Set discriminator_decay to 1e-6 (#145)
* Fix default value * Fix lint
1 parent 2a2a846 commit b5900f2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Diff for: ctgan/synthesizers/ctgan.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -133,8 +133,8 @@ class CTGANSynthesizer(BaseSynthesizer):
133133

134134
def __init__(self, embedding_dim=128, generator_dim=(256, 256), discriminator_dim=(256, 256),
135135
generator_lr=2e-4, generator_decay=1e-6, discriminator_lr=2e-4,
136-
discriminator_decay=0, batch_size=500, discriminator_steps=1, log_frequency=True,
137-
verbose=False, epochs=300, pac=10, cuda=True):
136+
discriminator_decay=1e-6, batch_size=500, discriminator_steps=1,
137+
log_frequency=True, verbose=False, epochs=300, pac=10, cuda=True):
138138

139139
assert batch_size % 2 == 0
140140

0 commit comments

Comments
 (0)