REx
REx copied to clipboard
[Feature Request]: Split ckpt into separate files
Content
currently, the dumped ckpt contains model_states, optimizer_state, and history. It would be better to split them into separate files and add configs to determine whether to dump each of these files.
Code of Conduct
- [x] I agree to follow this project's Code of Conduct