CSVLogger

torchrl.record.loggers.csv.CSVLogger(exp_name: str, log_dir: str | None = None, video_format: str = 'pt', video_fps: int = 30) → None

一个最小依赖项的 CSV 记录器。

参数:

• exp_name (str) – 实验的名称。

• log_dir (str 或 Path, 可选) – 实验应该保存的位置。默认为 <cur_dir>/csv_logs。

• video_format (str, 可选) – 调用 add_video() 时，视频应该如何保存。必须是 "pt" （视频保存为 video_<tag>_<step>.pt 文件，使用 torch.save）、"memmap" （视频保存为 video_<tag>_<step>.memmap 文件，使用 MemoryMappedTensor）、"mp4" （视频保存为 video_<tag>_<step>.mp4 文件，需要安装 torchvision）之一。默认为 "pt"。

• video_fps (int, 可选) – 如果 video_format=”mp4”，视频的帧率（帧/秒）。默认为 30。