reference/generated/torchrl.trainers.algorithms.configs.hooks.ClearCudaCacheConfig

Run in Google Colab

Colab

Download Notebook

Notebook

View on GitHub

GitHub

torchrl.trainers.algorithms.configs.hooks.ClearCudaCacheConfig

class torchrl.trainers.algorithms.configs.hooks.ClearCudaCacheConfig(interval: int = '???', _target_: str = 'torchrl.trainers.trainers.ClearCudaCache')

Configuration for the ClearCudaCache hook.

Examples

>>> from torchrl.trainers.algorithms.configs.hooks import ClearCudaCacheConfig
>>> from hydra.utils import instantiate
>>> hook = instantiate(ClearCudaCacheConfig(interval=100))