From 7c4356464c6f84aa38f276c92019a34c24566f60 Mon Sep 17 00:00:00 2001 From: Ludger Paehler Date: Tue, 17 Nov 2020 17:00:36 +0100 Subject: [PATCH] Minor typo in the description of Adam's beta 2 (#4715) Adam's beta 2 parameter was mistakenly referred to as the first order momentum of the gradient, whereas it should be the second order momentum. This has no effect on the correct working of the example. --- 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 8ebaad7965..088b625e31 100644 --- a/pl_examples/domain_templates/generative_adversarial_net.py +++ b/pl_examples/domain_templates/generative_adversarial_net.py @@ -76,7 +76,7 @@ class GAN(LightningModule): parser.add_argument("--b1", type=float, default=0.5, help="adam: decay of first order momentum of gradient") parser.add_argument("--b2", type=float, default=0.999, - help="adam: decay of first order momentum of gradient") + help="adam: decay of second order momentum of gradient") parser.add_argument("--latent_dim", type=int, default=100, help="dimensionality of the latent space")