DiskLogger

class torchtune.training.metric_logging.DiskLogger(log_dir: str, filename: Optional[str] = None, **kwargs)

将日志记录到磁盘。

参数：

• log_dir (str) – 存储日志的目录

• filename (可选[str]) – 可选的日志文件名。默认值：None，在这种情况下将使用 log_{unixtimestamp}.txt。

• **kwargs – 其他参数

警告

此记录器不是线程安全的。

注意

此记录器根据当前时间创建一个新文件。

close() → None

关闭日志资源，必要时刷新。在调用 close 后不应再写入日志。

log(name: str, data: Union[Tensor, ndarray, int, float], step: int) → None

记录标量数据。

参数：

• name (str) – 用于对标量进行分组的标签名称

• data (标量) – 要记录的标量数据

• step (int) – 要记录的步骤值

log_dict(payload: Mapping[str, Union[Tensor, ndarray, int, float]], step: int) → None

记录多个标量值。

参数：

• payload (映射[str, 标量]) – 标签名称和标量值的字典

• step (int) – 要记录的步骤值