Skip to content

fix: Fix activation checkpointing for mcore path#703

Merged
terrykong merged 1 commit intomainfrom
yifu/fix_act_ckpt
Jul 21, 2025
Merged

fix: Fix activation checkpointing for mcore path#703
terrykong merged 1 commit intomainfrom
yifu/fix_act_ckpt

Conversation

@yfw
Copy link
Contributor

@yfw yfw commented Jul 21, 2025

What does this PR do ?

Uses correct config for activation checkpointing. Thanks @guyueh1 for catching this!

Issues

List issues that this PR closes (syntax):

Usage

  • You can potentially add a usage example below
# Add a code snippet demonstrating how to use this

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you run the unit tests and functional tests locally? Visit our Testing Guide for how to run tests
  • Did you add or update any necessary documentation? Visit our Document Development Guide for how to write, build and test the docs.

Additional Information

  • ...

Signed-off-by: Yi-Fu Wu <yifu.wu@gmail.com>
@yfw yfw requested review from guyueh1 and terrykong July 21, 2025 19:34
@terrykong terrykong added this pull request to the merge queue Jul 21, 2025
Merged via the queue into main with commit 95452e1 Jul 21, 2025
13 of 14 checks passed
@terrykong terrykong deleted the yifu/fix_act_ckpt branch July 21, 2025 22:55
jialei777 pushed a commit to jialei777/nemo-rl that referenced this pull request Jul 23, 2025
Signed-off-by: Yi-Fu Wu <yifu.wu@gmail.com>
Signed-off-by: Jialei Chen <jialeic@google.com>
xxman-google pushed a commit to xxman-google/NeMo-RL that referenced this pull request Jul 28, 2025
KiddoZhu pushed a commit that referenced this pull request Jul 28, 2025
Signed-off-by: Yi-Fu Wu <yifu.wu@gmail.com>
rohitrango pushed a commit that referenced this pull request Jul 29, 2025
Signed-off-by: Yi-Fu Wu <yifu.wu@gmail.com>
xxman-google pushed a commit to xxman-google/NeMo-RL that referenced this pull request Jul 30, 2025
FannYYW pushed a commit to xxman-google/NeMo-RL that referenced this pull request Aug 5, 2025
Signed-off-by: Yi-Fu Wu <yifu.wu@gmail.com>
FannYYW pushed a commit to xxman-google/NeMo-RL that referenced this pull request Aug 5, 2025
soodoshll pushed a commit to soodoshll/RL that referenced this pull request Aug 13, 2025
Signed-off-by: Yi-Fu Wu <yifu.wu@gmail.com>
Signed-off-by: Qidong Su <qidongs@nvidia.com>
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.

3 participants