How to save the model? - Githubissues

prateekm08 / CycleGAN-straightforward

PyTorch implementation of CycleGAN.

14 stars 3 forks source link

Open jayanthsrinivas opened 5 years ago

jayanthsrinivas commented 5 years ago

Could you help me in saving the model.

spinachpasta commented 2 years ago

I'm currently trying to modify this notebook for latest pytorch and python3.8 I'm also implementing model save feature based on checkpoint functionality of pytorch. The following link might help you: https://pytorch.org/tutorials/recipes/recipes/saving_and_loading_a_general_checkpoint.html