I am a beginner in deep learning. I am trying to run a GAN (UNet based) model on the data - implemented in pytorch. but when I try to run the code. I get the following error :
**KeyError: ‘generatorX’   **
in the line:
generatorX.load_state_dict( checkpoint[‘generatorX’])
I am using the pretrained model / checkpoint (a file already provided by authors with .pth extension.
I cannot figure out what is ‘generatorX’. When I print the model I can see weights.
Any help will be highly appreciated.