prateekm08 / CycleGAN-straightforward

PyTorch implementation of CycleGAN.
14 stars 3 forks source link

How to save the model? #2

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