GAN Collection
pytorch-lightning implementations of GANs
Requirements
need for training
- torch==1.10.1
- torchvision==0.11.2
- pytorch-lightning==1.5.7
- tqdm==4.62.3
- wandb==0.12.10
- opencv-python-headless==4.5.5.62
- albumentations==1.1.0
- easydict==1.9
need for test
- pytest==7.0.0
- pytest-testmon==1.2.3
Training tutorial
python main.py \
--GAN {GAN_TYPE} \
--dataset {DATASET} \
--root_dir {DATASET_PATH} \
--image_channels 1 \
--gpus 1 \
--max_epochs 50Implementations
Please see the "Implementations"
Supported dataset
Please see the "Supported Dataset"