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')