GAN framework for 3D data
- pytorch >= 1.3
- lightning-pytorch >=0.5, recommend custom installation
- Check folder 'examples'. StyleGAN model can only be used for gpu training. You can change the noise generation on cpu for cpu traning.
- 'python train.py -h' for more options.