CountFramesLog

class torchrl.trainers.CountFramesLog(*args, **kwargs)

帧计数器钩子。

参数：:

• frame_skip (int) – 环境的帧跳跃。此参数对于跟踪总帧数（而不是可见帧数）非常重要。

• log_pbar (bool, 可选) – 如果为 True，奖励值将在进度条上记录。默认值为 False。

示例

>>> count_frames = CountFramesLog(frame_skip=frame_skip)
>>> trainer.register_op("pre_steps_log", count_frames)

register(trainer: Trainer, name: str = 'count_frames_log')

在训练器中的默认位置注册钩子。

参数：:

• trainer (Trainer) – 必须注册钩子的训练器。

• name (str) – 钩子的名称。

注意

要将钩子注册到默认位置以外的位置，请使用 register_op()。