Weights and Biases Configuration Logging

1

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

2

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
3

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

1 Like
4

now you can pass config parameter in WandbLogger.

5

When I run this I get the error:

    wandb_logger.experiment.config.update({"my_key": "my_value"})
AttributeError: 'function' object has no attribute 'update'.