From 43d9214569d59e794c7cf36d242a93b811088ad6 Mon Sep 17 00:00:00 2001 From: Daan Wynen Date: Mon, 30 Jan 2017 15:28:28 +0100 Subject: [PATCH] doc update was still using the wrong lr defaults --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 14a29d6..30826ca 100644 --- a/main.py +++ b/main.py @@ -26,8 +26,8 @@ parser.add_argument('--nz', type=int, default=100, help='size of the latent z ve parser.add_argument('--ngf', type=int, default=64) parser.add_argument('--ndf', type=int, default=64) parser.add_argument('--niter', type=int, default=25, help='number of epochs to train for') -parser.add_argument('--lrD', type=float, default=0.00005, help='learning rate for Critic, default=0.0002') -parser.add_argument('--lrG', type=float, default=0.00005, help='learning rate for Generator, default=0.0002') +parser.add_argument('--lrD', type=float, default=0.00005, help='learning rate for Critic, default=0.00005') +parser.add_argument('--lrG', type=float, default=0.00005, help='learning rate for Generator, default=0.00005') parser.add_argument('--beta1', type=float, default=0.5, help='beta1 for adam. default=0.5') parser.add_argument('--cuda' , action='store_true', help='enables cuda') parser.add_argument('--ngpu' , type=int, default=1, help='number of GPUs to use')