Shortcuts

WandbLogger

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

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.

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources