Commit a312e139 authored by sikhin.vc's avatar sikhin.vc

Update README.md

parent 85349f11
...@@ -31,7 +31,9 @@ $ lightweight_gan --data ./path/to/images --image-size 512 ...@@ -31,7 +31,9 @@ $ lightweight_gan --data ./path/to/images --image-size 512
``` ```
--data : Path to the folder where train images are present --data : Path to the folder where train images are present
--image-size : Size of the image to train --image-size : Size of the image to train
--batch-size : Reduce batch size if you are getting GPU error --batch-size : Reduce batch size if you are getting GPU error
While training the model, you can see generated images in results folder While training the model, you can see generated images in results folder
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment