Save final model from checkpoint file

agrimgupta92 / sgan

Code for "Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networks", Gupta et al, CVPR 2018

MIT License

826 stars 262 forks source link

Save final model from checkpoint file #106

Open JarlLemmens opened 2 years ago

JarlLemmens commented 2 years ago

Hi, I tried training a model for myself, but I noticed that the checkpoint files increment in size everytime that the checkpoint is saved. For the Zara1 model this became 25mb whereas the pretrained model is only 70kb. I was wondering if there is a way to extract this final model from the checkpoint file, or a separate way to save it?