Skip to content

Add support for setting WandB Entity / Comet Workspace via experiment config #3627

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

Open
wants to merge 7 commits into
base: main
Choose a base branch
from

Conversation

IQisMySenpai
Copy link

@IQisMySenpai IQisMySenpai commented Apr 4, 2025

This PR adds support for specifying the Weights & Biases (WandB) entity (team or organization) and the Comet workspace directly in the Nerfstudio experiment config.

Why?

Currently, users must either:

  • Rely on their default WandB entity / Comet Workspace, or
  • Move runs manually to the correct workspace after they have finished

This can be limiting when working across multiple teams or accounts. With this change, users can now explicitly set the WandB entity / Comet Workspace in the config through the workspace_name, making it easier to log runs under different teams or organizations without needing to modify global environment settings.

Notes

  • The entity / workspace must already exist in WandB / Comet.
  • If not specified, WandB / Comet will continue to fall back to the default entity (as it is currently), as defined in the user settings.

As this is my first PR to Nerfstudio, so I’m happy to take any suggestions or feedback!

@brentyi
Copy link
Collaborator

brentyi commented Apr 4, 2025

Is this argument useful for other writers, like Comet etc? If not, could we rename it to wandb_entity_name?

@IQisMySenpai
Copy link
Author

IQisMySenpai commented Apr 5, 2025

It seems comet has something similar to entities called workspace, which is currently also defaulted to the user's workspace. I'll do some more research for the other writers and integrate their different entity solutions.

Should I also migrate to the new comet api or would you prefer this to be a separate PR?

@IQisMySenpai IQisMySenpai changed the title Add support for setting WandB Entity via experiment config Add support for setting WandB Entity / Cursor Workspace via experiment config Apr 5, 2025
@IQisMySenpai IQisMySenpai changed the title Add support for setting WandB Entity / Cursor Workspace via experiment config Add support for setting WandB Entity / Comet Workspace via experiment config Apr 5, 2025
@IQisMySenpai IQisMySenpai force-pushed the integration-of-wandb-entity branch from ee88ef3 to a51eb26 Compare April 5, 2025 13:20
@IQisMySenpai IQisMySenpai force-pushed the integration-of-wandb-entity branch from f3f9ea9 to 9d98a82 Compare April 5, 2025 13:42
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