Issue loading a trained GAN when returning to work another day - CelebA Dataset

Hello. I would really love some help here. I am trying to return to work after training a WGAN model for nearly 7 hours -ish. And each time I try to return to work, I keep getting these same errors and I am having to restart the training process all over again.

Has someone found a fix to this problem? My current ipynb file is larger than 50 Meg and therefore isn’t getting uploaded on Github. So, here is the alternate link to my ipnyb file. https://drive.google.com/file/d/1ZqusC5q63ftc5Lhtl61FkbH69md6Hefl/view?usp=sharing

I would be extremely grateful to someone who can offer any advice!
Thank you in advance.

PS: I am using the latest version of fastai package. Running everything on Google Cloud.