Skip to content

Match no_save_optim configuration flag with no_load_optim flag.#239

Open
atoniolo76 wants to merge 3 commits into
mainfrom
alessio/match-optimizer-save-with-load-config
Open

Match no_save_optim configuration flag with no_load_optim flag.#239
atoniolo76 wants to merge 3 commits into
mainfrom
alessio/match-optimizer-save-with-load-config

Conversation

@atoniolo76

@atoniolo76 atoniolo76 commented Jul 10, 2026

Copy link
Copy Markdown
Contributor

no_save_optim=True can help prevent OOMs when dumping Adam optimizer states onto host RAM. no_load_optim is False by default and causes a Megatron error when no_save_optim=True.


Open in Devin Review

devin-ai-integration[bot]

This comment was marked as resolved.

…attribute setting from failing unconditionally.
devin-ai-integration[bot]

This comment was marked as resolved.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants