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。