fix demo
This commit is contained in:
parent
f14700a16a
commit
4c7c933326
|
@ -178,7 +178,6 @@ class GAN(pl.LightningModule):
|
|||
self.logger.experiment.add_image(f'generated_images', grid, self.current_epoch)
|
||||
|
||||
|
||||
|
||||
def main(hparams):
|
||||
|
||||
# ------------------------
|
||||
|
|
Loading…
Reference in New Issue