Skip to content

fix(sc): log config parameters to W&B - #3465

Merged
yuki-97 merged 1 commit into
mainfrom
ruit/fix-sc-wandb-config
Aug 3, 2026
Merged

fix(sc): log config parameters to W&B#3465
yuki-97 merged 1 commit into
mainfrom
ruit/fix-sc-wandb-config

Conversation

@RayenTian

@RayenTian RayenTian commented Aug 3, 2026

Copy link
Copy Markdown
Contributor

What does this PR do ?

Log the resolved SingleController master config as hyperparameters so W&B runs display config parameters in the Overview page, matching the existing training entrypoints.

Update SingleController test fixtures to use the real ClippedPGLossConfig, allowing the Pydantic master config to serialize correctly, and verify that hyperparameters are logged during actor initialization.

Issues

Before:

image

Now

image

Usage

Launch a SingleController recipe with W&B enabled. The resolved config parameters will be available in the W&B run Overview.

Before your PR is "Ready for review"

Pre checks:

  • Read and followed the contributor guidelines
  • Added/updated necessary tests
  • Ran unit and functional tests locally (not run per request)
  • Documentation changes are not required for this fix

Additional Information

git diff --check origin/main...HEAD passes.

@RayenTian
RayenTian requested review from a team as code owners August 3, 2026 06:16
@copy-pr-bot

copy-pr-bot Bot commented Aug 3, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@RayenTian

Copy link
Copy Markdown
Contributor Author

/ok to test 74988d5

Signed-off-by: ruit <ruit@nvidia.com>
@RayenTian
RayenTian force-pushed the ruit/fix-sc-wandb-config branch from 74988d5 to fb2c7eb Compare August 3, 2026 06:18
@RayenTian

Copy link
Copy Markdown
Contributor Author

/ok to test fb2c7eb

@RayenTian RayenTian added the CI:Lfast Runs a fast test suite and re-use nightly `main` container (but sync dependencies to PRs version) label Aug 3, 2026
@RayenTian
RayenTian requested a review from yuki-97 August 3, 2026 06:34

@yuki-97 yuki-97 left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

thanks for the fix!

@yuki-97
yuki-97 merged commit 89f4d1f into main Aug 3, 2026
93 checks passed
@yuki-97
yuki-97 deleted the ruit/fix-sc-wandb-config branch August 3, 2026 07:38
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CI:Lfast Runs a fast test suite and re-use nightly `main` container (but sync dependencies to PRs version)

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants