a yf @sddlZddlmZddlmZmZmZmZddlmZm Z m Z ddZ ddZ d d Z d d Zd dZddZddZddZddZe ddure e e eeeeeed niZdS)N)time) HUB_WEB_ROOTPREFIXHUBTrainingSessionevents)LOGGERRANKSETTINGScCs>tdvr:tddur:tdr:|jdur:t|jj|j|_dS)zDCreate a remote Ultralytics HUB session to log local model training.>rhubTZapi_keyN)rr hub_sessionrZcreate_sessionargsmodeltrainerr[/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/utils/callbacks/hub.pyon_pretrain_routine_start s&rcCs$t|dd}|r ttd|_dS)z6Logs info before starting timer for upload rate limit.r N)metricsckpt)getattrrtimersrsessionrrron_pretrain_routine_ends rcCst|dd}|ri|j|jdd|j}|jdkrPddlm}i|||}t||j |j<|j rv|j |j t |j d|jdkr|t |j d<i|_ dS)z;Uploads training progress metrics at the end of each epoch.r Ntrain)prefixr)model_info_for_loggersr)rZlabel_loss_itemsZtlossrepochZultralytics.utils.torch_utilsrjsondumpsZ metrics_queueZmetrics_upload_failed_queueupdaterr rate_limitsZupload_metrics)rrZ all_plotsrrrron_fit_epoch_ends"    r#cCsvt|dd}|rr|j|jk}t|jd|jdkrrttdt d|j j | |j |j|t|jd<dS)z8Saves checkpoints to Ultralytics HUB with rate limiting.r NrzUploading checkpoint z/models/)rZ best_fitnessZfitnessrrr"rinforrrid upload_modelrlast)rrZis_bestrrr on_model_save2s  r(cCsht|dd}|rdttd|j|j|j|jddddd|_ ttd td |j d dS) zIUpload final model and metrics to Ultralytics HUB at the end of training.r NzSyncing final model...zmetrics/mAP50-95(B)rT)mapfinalFu Done ✅ zView model at u 🚀) rrr$rr&rbestrgetaliveZ model_urlrrrr on_train_end>s  r.cCst|jdS)zRun events on train start.Nrr rrrron_train_startNsr0cCst|jdS)z Runs events on validation start.Nr/) validatorrrr on_val_startSsr2cCst|jdS)zRun events on predict start.Nr/)Z predictorrrron_predict_startXsr3cCst|jdS)zRun events on export start.Nr/)Zexporterrrron_export_start]sr4r T) rrr#r(r.r0r2r3r4)rrZultralytics.hubrrrrZultralytics.utilsrrr rrr#r(r.r0r2r3r4 callbacksrrrrs4