Thank you very much for reply. I run another file where model is trained from scratch.
I am getting the following errors: 'Caught RuntimeError in replica 0 on device 0’
in the line
fakeEnhanced = generatorX(realInput)
and the following error:
’Sizes of tensors must match except in dimension 2. Got 64 and 33 (The offending index is 0)'
where generator method is defined inside the model.py file in the line:
x6 = self.conv7(torch.cat([x5, x53_temp], dim=1))
I searched to correct the errors. Some people say that maybe there is some issue of DataParallel module and the original code was written to run on multiple GPUs.
Do  u have idea if it could be the case