a yf' @sddlZddlmZmZmZmZzXddlmZer6JeddusFJdaedZ ddl Z ddl m Z ddl mZmZWneeeefydZYn0dd d Zd d Zd dZddZddZddZereeeedniZdS)N)LOGGERSETTINGS TESTS_RUNNINGcolorstr) SummaryWriterZ tensorboardTz TensorBoard: )deepcopy) de_paralleltorchcCs(tr$|D]\}}t|||q dS)z"Logs scalar values to TensorBoard.N)WRITERitemsZ add_scalar)Zscalarsstepkvrc/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/utils/callbacks/tensorboard.py _log_scalarssrc Cs|jj}t|tr||fn|}t|j}tjddg|R|j |j d}t ^t j dtdt j dtjjdttX|jttjjt|j|ddgttdWd Wd d S1s0Yztt|j}||jdd }|D]}t|d rd |_ d |_!q||ttjj||ddgttdWn:ty}z t"td|WYd }~n d }~00Wd n1s0Yd S)zLog model graph to TensorBoard.)devicedtypeignore)categoryF)strictu#model graph visualization added ✅N)verboseexportTZ torchscriptu7WARNING ⚠️ TensorBoard graph visualization failure )#argsimgsz isinstanceintnextmodel parametersr Zzerosrrwarningscatch_warnings simplefilter UserWarningZjitZ TracerWarning contextlibsuppress Exceptionevalr Z add_graphtracerrinfoPREFIXrZfusemoduleshasattrrformatwarning)trainerrpZimr merrr_log_tensorboard_graph#s2    ,    r5c Csltrhz*tt|jattd|jdWn8tyf}z ttd|WYd}~n d}~00dS)z2Initialize TensorBoard logging with SummaryWriter.z!Start with 'tensorboard --logdir z!', view at http://localhost:6006/uLWARNING ⚠️ TensorBoard not initialized correctly, not logging this run. N) rstrsave_dirr rr+r,r(r0)r1r4rrron_pretrain_routine_startFs r8cCstr t|dS)zLog TensorBoard graph.N)r r5r1rrron_train_startQsr:cCs2t|j|jdd|jdt|j|jddS)z6Logs scalar statistics at the end of a training epoch.train)prefixrN)rZlabel_loss_itemsZtlossepochlrr9rrron_train_epoch_endWsr?cCst|j|jddS)z,Logs epoch metrics at end of training epoch.rN)rZmetricsr=r9rrron_fit_epoch_end]sr@)r8r:r@r?)r)r&Zultralytics.utilsrrrrZtorch.utils.tensorboardrr r,r"copyrZultralytics.utils.torch_utilsrr ImportErrorAssertionError TypeErrorAttributeErrorrr5r8r:r?r@ callbacksrrrrs6    #