fix model arguements (#5653)

2021-01-28 09:56:30 +09:00 · 2021-01-28 09:56:30 +09:00 · d1200af358
parent 230f2b723c
commit d1200af358
1 changed files with 1 additions and 1 deletions
--- a/pl_examples/domain_templates/generative_adversarial_net.py
+++ b/pl_examples/domain_templates/generative_adversarial_net.py
@ -240,7 +240,7 @@ def main(args: Namespace) -> None:
    # ------------------------
    # 1 INIT LIGHTNING MODEL
    # ------------------------
-    model = GAN(args)
+    model = GAN(lr=args.lr, b1=args.b1, b2=args.b2, latent_dim=args.latent_dim)

    # ------------------------
    # 2 INIT TRAINER