Revert "Add L2 weight penalty to descriminator"

This reverts commit f99b90b.
hellomoto-ai · May 11, 2019 · 48b3a7f · 48b3a7f
1 parent b283b20
commit 48b3a7f
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/train_vae_gan.py b/train_vae_gan.py
@@ -91,8 +91,7 @@ def _get_trainer(args):
     optimizers = {
         'encoder': opt(model.vae.encoder.parameters(), lr=args.lr),
         'decoder': opt(model.vae.decoder.parameters(), lr=args.lr),
-        'discriminator': opt(
-            model.discriminator.parameters(), lr=args.lr, weight_decay=1e-5),
+        'discriminator': opt(model.discriminator.parameters(), lr=args.lr),
     }
     trainer = sp_vae_gan.trainer.Trainer(
         model, optimizers, train_loader, test_loader, device, args.output_dir,