DiskLogger

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

磁盘记录器。

参数:

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

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

• **kwargs – 附加参数

警告

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

注意

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

close() → None

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

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

记录标量数据。

参数:

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

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

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

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

记录多个标量值。

参数:

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

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