快捷方式

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 (strPath, 可选) – 实验应保存到的位置。默认为 <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。

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源