LogScalar

class torchrl.trainers.LogScalar(logname='r_training', log_pbar: bool = False, reward_key: Optional[Union[str, tuple]] = None)

獎勵記錄器鉤子。

引數：

• logname (str, 可選) – 要記錄的獎勵名稱。預設值為 "r_training"。

• log_pbar (bool, 可選) – 如果為 True，獎勵值將記錄在進度條上。預設值為 False。

• reward_key (str 或 tuple, 可選) – 在輸入 batch 中查詢獎勵的鍵。預設為 ("next", "reward")

示例

>>> log_reward = LogScalar(("next", "reward"))
>>> trainer.register_op("pre_steps_log", log_reward)

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

在訓練器的預設位置註冊鉤子。

引數：

• 訓練器 (Trainer) – 必須註冊鉤子的訓練器。

• 名稱 (str) – 鉤子的名稱。

注意

要在預設位置以外的其他位置註冊鉤子，請使用 register_op()。