Hey, I think I’m wrong.
Actually the when I did it again, the training was finished. And the first time the training was interrupted and did complete.
I still don’t know exactly why is that. The code used torch.nn.DataParallel for both Generator and Discriminator during training. And I loaded the weights to gpu while testing.
How do I check if weights are loaded?