快捷方式

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

文档

访问 PyTorch 的完整开发者文档

查看文档

教程

获取面向初学者和高级开发者的深度教程

查看教程

资源

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

查看资源