-
Notifications
You must be signed in to change notification settings - Fork 2.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Fix export of configuration parameters to Weights and Biases #10995
Conversation
a81f622
to
26fa93a
Compare
Add test to ensure json serializability of hparams_init after restore Signed-off-by: Sam Oluwalana <[email protected]>
26fa93a
to
c8438f9
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Makes sense, thanks for the fix !
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. Thanks!
@ericharper @titu1994 can you trigger the Nemo_CICD_Test, cicd-test-container-setup, and gpu-test actions? |
Head branch was pushed to by a user without write access
Signed-off-by: Sam Oluwalana <[email protected]>
f8856fa
to
46e5343
Compare
Signed-off-by: soluwalana <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM. Thanks!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks !
…10995) * Fix export of configuration parameters to weights and biases Add test to ensure json serializability of hparams_init after restore Signed-off-by: Sam Oluwalana <[email protected]> * Function was moved in recent versions Signed-off-by: Sam Oluwalana <[email protected]> * Apply isort and black reformatting Signed-off-by: soluwalana <[email protected]> --------- Signed-off-by: Sam Oluwalana <[email protected]> Signed-off-by: soluwalana <[email protected]> Co-authored-by: soluwalana <[email protected]>
…10995) * Fix export of configuration parameters to weights and biases Add test to ensure json serializability of hparams_init after restore Signed-off-by: Sam Oluwalana <[email protected]> * Function was moved in recent versions Signed-off-by: Sam Oluwalana <[email protected]> * Apply isort and black reformatting Signed-off-by: soluwalana <[email protected]> --------- Signed-off-by: Sam Oluwalana <[email protected]> Signed-off-by: soluwalana <[email protected]> Co-authored-by: soluwalana <[email protected]>
* Fix export of configuration parameters to weights and biases Add test to ensure json serializability of hparams_init after restore Signed-off-by: Sam Oluwalana <[email protected]> * Function was moved in recent versions Signed-off-by: Sam Oluwalana <[email protected]> * Apply isort and black reformatting Signed-off-by: soluwalana <[email protected]> --------- Signed-off-by: Sam Oluwalana <[email protected]> Signed-off-by: soluwalana <[email protected]> Co-authored-by: soluwalana <[email protected]>
…10995) * Fix export of configuration parameters to weights and biases Add test to ensure json serializability of hparams_init after restore Signed-off-by: Sam Oluwalana <[email protected]> * Function was moved in recent versions Signed-off-by: Sam Oluwalana <[email protected]> * Apply isort and black reformatting Signed-off-by: soluwalana <[email protected]> --------- Signed-off-by: Sam Oluwalana <[email protected]> Signed-off-by: soluwalana <[email protected]> Co-authored-by: soluwalana <[email protected]>
What does this PR do ?
This PR corrects an error in ModelPT.py that only partially converts the _hparams_initial from an OmegaConf object to a dictionary before exporting to the various loggers.
This was particularly an issue with search functionality based on metadata in Weights and Biases due to the cfg object being a string representation of the OmegaConf object.
Collection: Core
Changelog
Reference Images
Prior to fix
cfg is recorded as a str(OmegaConf.DictConfig) in Weights and Biases
After fix
cfg is correctly searchable in Weights and Biases
GitHub Actions CI
The Jenkins CI system has been replaced by GitHub Actions self-hosted runners.
The GitHub Actions CI will run automatically when the "Run CICD" label is added to the PR.
To re-run CI remove and add the label again.
To run CI on an untrusted fork, a NeMo user with write access must first click "Approve and run".
Before your PR is "Ready for review"
Pre checks:
PR Type:
If you haven't finished some of the above items you can still open "Draft" PR.
Who can review?
Anyone in the community is free to review the PR once the checks have passed.
Contributor guidelines contains specific people who can review PRs to various areas.
Additional Information