How to save checkpoint? #2236
Unanswered
yhcc
asked this question in
Community | Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
From the tutorial, it shows to use save_checkpoint function to save state, but
ColossalAI/colossalai/utils/checkpointing.py
Lines 184 to 188 in 31fe842
the optimizer and scheduler are not saved. What is the right way to checkpoint so that I can resume training? Thx~
Beta Was this translation helpful? Give feedback.
All reactions