diff --git a/examples/domain_templates/gan.py b/examples/domain_templates/gan.py index 8a31b74652..5a94fcd4bc 100644 --- a/examples/domain_templates/gan.py +++ b/examples/domain_templates/gan.py @@ -178,7 +178,6 @@ class GAN(pl.LightningModule): self.logger.experiment.add_image(f'generated_images', grid, self.current_epoch) - def main(hparams): # ------------------------