I have a model with a GAN architecture trained in a kaggle notebook and I want save it and use it in my local environment.The model is trained using GPU P100 but my local machine does not support GUP.
Following code has used to save the model usinf torch.