Weights and Biases Configuration Logging

Is there way to pass configurations dictionary to Weights and Biases logger? https://pytorch-lightning.readthedocs.io/en/latest/loggers.html#weights-and-biases.

I don’t believe it is directly supported but you can access the wandb object through the .experiment property:

from pytorch_lightning.loggers import WandbLogger
wandb_logger = WandbLogger()
wandb_logger.experiment.config.update({"my_key": "my_value"})

Feel free to open a PR (or at least an issue) if you would like to add this functionality to the WandbLogger :slight_smile:

2 Likes

yeah an additional parameter config in WandbLogger would do the job.

1 Like

now you can pass config parameter in WandbLogger.