No batch normalization for the critic.

rcamino · Aug 20, 2018 · 5e6146d · 5e6146d
1 parent 9dc0287
commit 5e6146d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/multi_categorical_gans/methods/arae/trainer.py b/multi_categorical_gans/methods/arae/trainer.py
@@ -394,7 +394,7 @@ def main():
     discriminator = Discriminator(
         options.code_size,
         hidden_sizes=parse_int_list(options.discriminator_hidden_sizes),
-        bn_decay=options.bn_decay,
+        bn_decay=0,  # no batch normalization for the critic
         critic=True
     )
 

diff --git a/multi_categorical_gans/methods/mc_wgan_gp/trainer.py b/multi_categorical_gans/methods/mc_wgan_gp/trainer.py
@@ -276,7 +276,7 @@ def main():
     discriminator = Discriminator(
         features.shape[1],
         hidden_sizes=parse_int_list(options.discriminator_hidden_sizes),
-        bn_decay=options.bn_decay,
+        bn_decay=0,  # no batch normalization for the critic
         critic=True
     )