Rate this Page

torchrl.trainers.algorithms.configs.hooks.CountFramesLogConfig#

class torchrl.trainers.algorithms.configs.hooks.CountFramesLogConfig(frame_skip: int = 1, log_pbar: bool = False, _target_: str = 'torchrl.trainers.trainers.CountFramesLog')[source]#

Configuration for the CountFramesLog hook.

Examples

>>> from torchrl.trainers.algorithms.configs.hooks import CountFramesLogConfig
>>> from hydra.utils import instantiate
>>> hook = instantiate(CountFramesLogConfig(frame_skip=4))