TensorDictRecorder

torchrl.record.TensorDictRecorder(out_file_base: str, skip_reset: bool = True, skip: int = 4, in_keys: Optional[Sequence[str]] = None) → None

TensorDict 记录器。

当调用 ‘dump’ 方法时，这个类将保存一个 tensordict 栈，来自于 env.step(td) 的结果，保存到一个文件中，该文件的前缀由 out_file_base 参数定义。

参数:

• out_file_base (str) – 一个字符串，定义了将写入 tensordict 的文件前缀。

• skip_reset (bool) – 如果为 True，列表中的第一个 TensorDict 将被丢弃 (通常是调用 env.reset() 产生的 tensordict) 默认值: True

• skip (int) – 保存的 tensordict 的帧间隔。默认值: 4