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”，则表示视频的帧率（frames-per-seconds）。默认为 30。