CSVLogger¶
- torchrl.record.loggers.csv.CSVLogger(exp_name: str, log_dir: str | None = None, video_format: str ='pt', video_fps: int =30) None [source]¶
一个依赖最小的 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。