Revert "Add L2 weight penalty to descriminator"

This reverts commit f99b90b.

Author: moto-hellomoto-ai

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,