Skip to content

More Flexible WandB Interface#80

Open
AntoineRichard wants to merge 1 commit intoleggedrobotics:mainfrom
AntoineRichard:antoine/wandb_update
Open

More Flexible WandB Interface#80
AntoineRichard wants to merge 1 commit intoleggedrobotics:mainfrom
AntoineRichard:antoine/wandb_update

Conversation

@AntoineRichard
Copy link
Copy Markdown

Hey there!

This is a simple PR that brings a bit more flexibility when using a WandB logger.
It's following the same logic as SKRL's implementation.

I.e. instead of passing:

wandb_project = "Whatever"

We can now use:

 wandb_kwargs = {
        "project": "jetbot_direct",
        "entity": "spacer-rl",
        "group": "zeroG",
    } 

I'm happy to make any modification, like making it backward compatible with the current default config style.

LMK,

Antoine

@AntoineRichard AntoineRichard changed the title Changed the way WandB is being initialized. It should be more flexible. More Flexible WandB Interface Apr 1, 2025

try:
project = cfg["wandb_project"]
project = cfg["wandb_kwargs"]["project"]
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

Could you also modify the example_config.yaml to reflect the change?

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