Ray Tune

W&B integrates with Ray by providing a logger for use with RLlib or Tune runs.


from ray import tune
from wandb.ray import WandbLogger
tune.run("PG", loggers=[WandbLogger], config={
"monitor": True, "env_config": {
"wandb": {"project": "my-project-name", "monitor_gym": True}}})

The "wandb" dictionary of env_config will be passed to wandb.init when training starts. See the wandb.init docs for all available options.