Merge pull request pytorch#970 from pytorch/ndf

updated dcgan docs
mpc001 · Mar 9, 2022 · 1ff11db · 1ff11db
2 parents 886b74e + 6a0582a
commit 1ff11db
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dcgan/README.md b/dcgan/README.md
@@ -31,8 +31,8 @@ optional arguments:
   --batchSize BATCHSIZE input batch size
   --imageSize IMAGESIZE the height / width of the input image to network
   --nz NZ               size of the latent z vector
-  --ngf NGF
-  --ndf NDF
+  --ngf NGF             number of filters in the generator
+  --ndf NDF             number of filters in the discriminator
   --niter NITER         number of epochs to train for
   --lr LR               learning rate, default=0.0002
   --beta1 BETA1         beta1 for adam. default=0.5