o W h @sddlZddlmZddlmZmZmZmZddlmZm Z m Z ddZ ddZ d d Z d d Zd dZddZddZddZddZe ddurXe e e eeeeeed ZdSiZdS)N)time) HUB_WEB_ROOTPREFIXHUBTrainingSessionevents)LOGGERRANKSETTINGScCsNtdvrtddur!tdr#|jdur%t|jj|j|_dSdSdSdSdS)zDCreate a remote Ultralytics HUB session to log local model training.>rhubTapi_keyN)rr hub_sessionrcreate_sessionargsmodeltrainerrS/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/utils/callbacks/hub.pyon_pretrain_routine_start s&rcCs(t|dd}rttd|_dSdS)zBInitialize timers for upload rate limiting before training begins.r N)metricsckpt)getattrrtimersrsessionrrron_pretrain_routine_endsrcCst|dd}rWi|j|jdd|j}|jdkr(ddlm}i|||}t||j |j<|j r;|j |j t |j d|jdkrY|t |j d<i|_ dSdSdS)zMUpload training progress metrics to Ultralytics HUB at the end of each epoch.r Ntrain)prefixr)model_info_for_loggersr)rlabel_loss_itemstlossrepochultralytics.utils.torch_utilsrjsondumps metrics_queuemetrics_upload_failed_queueupdaterr rate_limitsupload_metrics)rr all_plotsrrrron_fit_epoch_ends$    r,cCs~t|dd}r;|j|jk}t|jd|jdkr=ttdt d|j j | |j |j|t|jd<dSdSdS)z?Upload model checkpoints to Ultralytics HUB with rate limiting.r NrzUploading checkpoint z/models/)r best_fitnessfitnessrrr)rinforrrid upload_modelr"last)rris_bestrrr on_model_save0s r4cCslt|dd}r4ttd|j|j|j|jddddd|_ ttd td |j d dSdS) 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/rr1r"bestrgetalive model_urlrrrr on_train_end;s "r;cCt|jdS)zRun events on train start.Nrrrrrron_train_startJr>cCr<)zRun events on validation start.Nr=) validatorrrr on_val_startOr?rAcCr<)zRun events on predict start.Nr=) predictorrrron_predict_startTr?rCcCr<)zRun events on export start.Nr=)exporterrrron_export_startYr?rEr T) rrr,r4r;r>rArCrE)r$rultralytics.hubrrrrultralytics.utilsrrr rrr,r4r;r>rArCrE callbacksrrrrs6