Shortcuts

Loggers

Logger classes for experiment tracking and visualization.

Logger(exp_name, log_dir)

A template for loggers.

csv.CSVLogger(exp_name[, log_dir, ...])

A minimal-dependency CSV logger.

mlflow.MLFlowLogger(exp_name, tracking_uri)

Wrapper for the mlflow logger.

tensorboard.TensorboardLogger(exp_name[, ...])

Wrapper for the Tensoarboard logger.

wandb.WandbLogger(*args, **kwargs)

Wrapper for the wandb logger.

get_logger(logger_type, logger_name, ...)

Get a logger instance of the provided logger_type.

generate_exp_name(model_name, experiment_name)

Generates an ID (str) for the described experiment using UUID and current date.

Recording utils

VideoRecorder(logger, tag[, in_keys, skip, ...])

Video Recorder transform.

TensorDictRecorder(out_file_base[, ...])

TensorDict recorder.

PixelRenderTransform([out_keys, preproc, ...])

A transform to call render on the parent environment and register the pixel observation in the tensordict.

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