I made a new open source tool called JellyML that lets you go back to any of your checkpoints, and reproduce your code exactly as it was when you trained it.
You can find the website here:
The GitHub repo:
https://gitHub.com/mmulet/jellyml
You can install it with pip:
pip install jellyml