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