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()。