From d1200af358f074b8afa8ace8c5ce2b4619634a63 Mon Sep 17 00:00:00 2001 From: Jihoon Lee Date: Thu, 28 Jan 2021 09:56:30 +0900 Subject: [PATCH] fix model arguements (#5653) --- pl_examples/domain_templates/generative_adversarial_net.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pl_examples/domain_templates/generative_adversarial_net.py b/pl_examples/domain_templates/generative_adversarial_net.py index e4a9b15dbd..2aa2a9f73d 100644 --- 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