Are there any codes for training GANs?

[DreamtaleCore]

Great job! Are there some demos or code snippets for training a GAN? Many thanks

[ashleve]

I don't have a GAN example with the template but here is an official lightning tutorial: https://pytorch-lightning.readthedocs.io/en/stable/notebooks/lightning_examples/basic-gan.html

I believe you could train it with src/train.py without changes if you just implement a LightningModule and LightningDataModule from that tutorial.