WandbLogger

torchrl.record.loggers.wandb.WandbLogger(*args, **kwargs)

Wrapper for the wandb logger.

The keyword arguments are mainly based on the wandb.init() kwargs. See the doc here.

Parameters:

• exp_name (str) – The name of the experiment.

• offline (bool, optional) – if True, the logs will be stored locally only. Defaults to False.

• save_dir (path, optional) – the directory where to save data. Exclusive with log_dir.

• log_dir (path, optional) – the directory where to save data. Exclusive with save_dir.

• id (str, optional) – A unique ID for this run, used for resuming. It must be unique in the project, and if you delete a run you can’t reuse the ID.

• project (str, optional) – The name of the project where you’re sending the new run. If the project is not specified, the run is put in an "Uncategorized" project.

Keyword Arguments:

• fps (int, optional) – Number of frames per second when recording videos. Defaults to 30.

• **kwargs – Extra keyword arguments for wandb.init. See relevant page for more info.