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" (视频以 torch.save 格式保存为 video_<tag>_<step>.pt 文件)、"memmap" (视频以 MemoryMappedTensor 格式保存为 video_<tag>_<step>.memmap 文件)、"mp4" (视频保存为 video_<tag>_<step>.mp4 文件，需要安装 torchvision)。默认为 "pt"。

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