TensorDictRecorder

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

TensorDict 录制器。

当调用“dump”方法时，此类会将 env.step(td) 产生的 tensordict 堆栈保存在一个文件中，该文件的前缀由 out_file_base 参数定义。

参数:

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

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

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