torch.monitor 的源代码

从 typing 导入 TYPE_CHECKING 从 torch._C._monitor 导入* # noqa: F403 从 torch._C._monitor 导入_WaitCounter, _WaitCounterTracker if TYPE_CHECKING: 从 torch.utils.tensorboard 导入 SummaryWriter STAT_EVENT = "torch.monitor.Stat"
[文档]class TensorboardEventHandler:
"""
TensorboardEventHandler 是一个事件处理器，它将已知事件写入
提供的 SummaryWriter。

目前仅支持 ``torch.monitor.Stat`` 事件记录
以标量形式。

示例：
>>> # xdoctest: +REQUIRES(env:TORCH_DOCTEST_MONITOR)
>>> # xdoctest: +REQUIRES(module:tensorboard)
>>> 从 torch.utils.tensorboard 导入 SummaryWriter
>>> 从 torch.monitor 导入 TensorboardEventHandler，register_event_handler
>>> writer = SummaryWriter("log_dir")
>>> 注册事件处理器(TensorboardEventHandler(writer))
"""

[文档]    def __init__(self, writer: "SummaryWriter") -> None:
"""
构建一个 ``TensorboardEventHandler``。
"""
self._writer = writer

    def __call__(self, event: Event) -> None:
        if event.name == STAT_EVENT:
            for k, v in event.data.items():
                self._writer.add_scalar(k, v, walltime=event.timestamp.timestamp())