Skip to content

Async checkpoint saving in GRPO loop #1471

@guyueh1

Description

@guyueh1

Is your feature request related to a problem? Please describe.
Currently async_save is disabled in mcore path checkpoint, serialization takes a long time with training paused; should test async_save and resolve any issues coming up.

Describe the solution you'd like
A clear and concise description of what you want to happen.

Describe alternatives you've considered
A clear and concise description of any alternative solutions or features you've considered.

Additional context
Add any other context or screenshots about the feature request here.

Metadata

Metadata

Assignees

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions