Dead Kernel warning in Pix2Pix Assignment

In cell after “Transform”, where we initialize the generator and the discriminator, I keep getting a pop up dialog which says: “Kernel has died, will restart automatically”. Please see the attached screenshot and let me know how to overcome this issue. Thanks!

I was able to pass the assignment by changing pretrained to False

Hi @manav9899,
I have raised a ticket to fix the loading of the pretrained checkpoint. We will get back to you as soon as it is resolved.

1 Like

Hi @manav9899

Please add the parameter map_location = device while loading the pre-trained checkpoint. Check Line#16 in the following screenshot.

Do let us know if you still face any issues. :slight_smile:

1 Like