Hi, I trained a GAN on a dataset and then saved it using torch.save() . However when I load this saved model and use the generator to generate results, they are completely meaningless.
While loading I am using
generator.load_state_dict(path['basic_celeba_gen'], strict = False)

I had to set this flag to “False” because while training the generator I am using a Scheduler that updates the weights at certain epochs and if I don’t use the False flag, I am getting an error. However, I think this is what is causing the generation to fail after loading the model.
How do I fix this?