Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix model arguements (#5653)

This commit is contained in:
Jihoon Lee 2021-01-28 09:56:30 +09:00 committed by Jirka Borovec
parent 230f2b723c
commit d1200af358
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pl_examples/domain_templates/generative_adversarial_net.py
View File

@ -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