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。