TensorDictRecorder

torchrl.record.TensorDictRecorder(out_file_base: str, skip_reset: bool = True, skip: int = 4, in_keys: Optional[Sequence[str]] = 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