
    QVji                     X    d dl mZ d dlT d dlmZmZ erd dlmZ dZ G d d          ZdS )	    )TYPE_CHECKING)*)_WaitCounter_WaitCounterTracker)SummaryWriterztorch.monitor.Statc                   *    e Zd ZdZd	dZdeddfdZdS )
TensorboardEventHandlera]  
    TensorboardEventHandler is an event handler that will write known events to
    the provided SummaryWriter.

    This currently only supports ``torch.monitor.Stat`` events which are logged
    as scalars.

    Example:
        >>> # xdoctest: +REQUIRES(env:TORCH_DOCTEST_MONITOR)
        >>> # xdoctest: +REQUIRES(module:tensorboard)
        >>> from torch.utils.tensorboard import SummaryWriter
        >>> from torch.monitor import TensorboardEventHandler, register_event_handler
        >>> writer = SummaryWriter("log_dir")
        >>> register_event_handler(TensorboardEventHandler(writer))
    writerr   returnNc                     || _         dS )z=
        Constructs the ``TensorboardEventHandler``.
        N)_writer)selfr
   s     P/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/monitor/__init__.py__init__z TensorboardEventHandler.__init__   s         eventc                     |j         t          k    rS|j                                        D ];\  }}| j                            |||j                                                   :d S d S )N)walltime)name
STAT_EVENTdataitemsr   
add_scalar	timestamp)r   r   kvs       r   __call__z TensorboardEventHandler.__call__$   st    :##
((** T T1''1u7P7P7R7R'SSSS $#T Tr   )r
   r   r   N)__name__
__module____qualname____doc__r   Eventr    r   r   r	   r	      s\             Te T T T T T T Tr   r	   N)	typingr   torch._C._monitorr   r   torch.utils.tensorboardr   r   r	   r#   r   r   <module>r'      s                    ? ? ? ? ? ? ? ?  6555555!
T T T T T T T T T Tr   