Skip to content

Latest commit

 

History

History

01_lightweight_gan

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 

Training and Using the Lightweight GAN

The easiest part of the pipeline that just so happens to take the longest.

For me, I ran the following command:

lightweight_gan --data /home/nathancooperjones/GitHub/apebase/ipfs_jpg/ --name lightweight_ape_gan --image-size 512 --batch-size 16 --gradient-accumulate-every 4 --num-train-steps 300000 --aug-prob 0.35 --aug-types [cutout,color] --attn-res-layers [32,64]

If you instead would like to start with my pre-trained weights, you can download those here. With these weights, you can follow the directions on lucidrains' source repo here to generate images, with something similar to:

lightweight_gan --name lightweight_ape_gan --load-from 256 --generate --num-image-tiles 5

Good luck!