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。