Issue in saving checkpoints

Let me know if the answer here clears up your question.