WandbLogger

torchrl.record.loggers.wandb.WandbLogger(*args, **kwargs)

Wandb 日志记录器的包装器。

关键字参数主要基于 wandb.init() 的关键字参数。请参见 此处 的文档。

参数:

• exp_name (str) – 实验的名称。

• offline (bool, optional) – 如果为 True，则日志仅存储在本地。默认为 False。

• save_dir (path, optional) – 保存数据的目录。与 log_dir 互斥。

• log_dir (path, optional) – 保存数据的目录。与 save_dir 互斥。

• id (str, optional) – 此运行的唯一 ID，用于恢复。它必须在项目中是唯一的，并且如果您删除一个运行，则无法重用该 ID。

• project (str, optional) – 您发送新运行的项目名称。如果未指定项目，则该运行将被放入一个名为 "Uncategorized" 的项目中。

关键字参数:

• fps (int, optional) – 录制视频时的每秒帧数。默认为 30。

• **kwargs – wandb.init 的额外关键字参数。有关更多信息，请参阅相关页面。