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)
|
self.logger.experiment.add_image(f'generated_images', grid, self.current_epoch)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def main(hparams):
|
def main(hparams):
|
||||||
|
|
||||||
# ------------------------
|
# ------------------------
|
||||||
|
|
Loading…
Reference in New Issue