CountFramesLog

类 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: 训练器, name: str = 'count_frames_log')

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

参数：

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

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

注意

要在非默认位置注册钩子，请使用 register_op()。