reference/generated/torchrl.render.backends.RenderBackend

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

RenderBackend

class torchrl.render.backends.RenderBackend(*args, **kwargs)

Protocol implemented by rlrender frame-capture backends.

Examples

>>> from torchrl.render.backends import NullRenderBackend
>>> backend = NullRenderBackend()
>>> backend.name
'null'

capture(env: Any, tensordict: TensorDictBase, config: RenderConfig, *, step: int, trajectory_index: int) → FrameBundle | None

Captures frames for one rollout step.

close() → None

Closes backend resources.

supports(env: Any, config: RenderConfig) → bool

Returns whether the backend can capture from this environment.