Save final model from checkpoint file

[JarlLemmens]

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?