o
Xεi
@ s4 d dl Z d dlmZ d dlmZ G dd deZdS ) N)
SummaryWriter)BaseDashboardLoggerc @ s e Zd Zdd Zdd Zdd Zdeded ed
d fd d
Z dd Z
dd Z dd Z d ddZ
dd Zdd Zdd Zd d Zd d Zd S )!TensorboardLoggerc C s || _ t|| _d S N)
model_namer writer)selflog_dirr r
V/home/ubuntu/.local/lib/python3.10/site-packages/trainer/logging/tensorboard_logger.py__init__ s zTensorboardLogger.__init__c C s d}| D ]n\}}| dkr | jd||| | nQ| jd||| | | jd||| | | jd||| | | jd||| | | j d|||| | j d|||j
| |d7 }qd S ) N zlayer{}-{}/valuezlayer{}-{}/maxzlayer{}-{}/minzlayer{}-{}/meanzlayer{}-{}/stdzlayer{}-{}/paramzlayer{}-{}/grad) named_parametersnumelr
add_scalarformatmaxminmeanstd
add_histogramgrad)r modelstep layer_numnameparamr
r
r
model_weights
s
z TensorboardLogger.model_weightsc C s | dd| dd d S )Nz model-configzz
r )add_textto_json)r configr
r
r
add_config s z TensorboardLogger.add_configtitlevaluer returnNc C | j ||| d S r )r r )r r" r# r r
r
r r z TensorboardLogger.add_scalarc C s | j j||||d d S )N sample_rate)r add_audio)r r" audior r( r
r
r r) ! s zTensorboardLogger.add_audioc C r% r )r r )r r" textr r
r
r r $ r& zTensorboardLogger.add_textc C r% r )r
add_figure)r r" figurer r
r
r r, ' r& z TensorboardLogger.add_figurec c s d V d S r r
)r file_or_dirr
artifact_typealiasesr
r
r add_artifact* s
z TensorboardLogger.add_artifactc C s, | D ]\}}| d|||| qd S N{}/{})itemsr r )r
scope_namescalarsr keyr# r
r
r add_scalars- s z TensorboardLogger.add_scalarsc C s. | D ]\}}| jd|||| qd S r2 )r4 r r, r )r r5 figuresr r7 r# r
r
r add_figures1 s z TensorboardLogger.add_figuresc C sb | D ]*\}}|jdkr|d}z| jd|||||d W q ty. t Y qw d S )Nfloat16float32r3 r' )r4 dtypeastyper) r RuntimeError traceback print_exc)r r5 audiosr r( r7 r# r
r
r
add_audios5 s
z TensorboardLogger.add_audiosc C | j d S r )r flushr r
r
r rE C zTensorboardLogger.flushc C rD r )r closerF r
r
r finishF rG zTensorboardLogger.finishr )__name__
__module__ __qualname__r r r! strfloatintr r) r r, r1 r8 r: rC rE rI r
r
r
r r s
r )r@ torch.utils.tensorboardr trainer.logging.base_dash_loggerr r r
r
r
r s