Load only Generator for synthesis

[mcomunita]

Hi there,

The suggested way to load a checkpoint is:
model = MyLightningModule.load_from_checkpoint(PATH)

When training a GAN, we obviously have 2 models being saved.

How can I only resume the Generator?

Thanks