快捷方式

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

文档

获取 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

查找开发资源并获得问题解答

查看资源