j ddlZddlmZddlmZmZmZddlmZmZm Z ddl m Z dZ dZ dZd Zd Zd Zd Zd ZdZe ddur e e eeeeeeed niZdS)N)time) HUB_WEB_ROOTPREFIXHUBTrainingSession)LOGGERRANKSETTINGS)eventsctdvrNtddurAtdr6|j1tj|jj|j|_dSdSdSdSdS)zDCreate a remote Ultralytics HUB session to log local model training.>rhubTapi_keyN)rr hub_sessionrcreate_sessionargsmodeltrainers d/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/utils/callbacks/hub.pyon_pretrain_routine_startr sh w8E?d22x 7J2wObOj0? @RT[T`aa2222OjOjctt|ddx}r$ttd|_dSdS)zBInitialize timers for upload rate limiting before training begins.rN)metricsckpt)getattrrtimersrsessions ron_pretrain_routine_endrs@'=$777w=%)VVTVV<<==rct|ddx}ri||jd|j}|jdkrddlm}i|||}tj||j |j<|j r|j |j t|j dz |jdkr5|t|j d<i|_ dSdSdS)zMUpload training progress metrics to Ultralytics HUB at the end of each epoch.rNtrain)prefixr)model_info_for_loggersr)rlabel_loss_itemstlossrepochultralytics.utils.torch_utilsr#jsondumps metrics_queuemetrics_upload_failed_queueupdaterr rate_limitsupload_metrics)rr all_plotsr#s ron_fit_epoch_endr0s,'=$777w' &&w}W&EE o  =A   L L L L L LH9H(>(>w(G(GHI/3z)/D/Dgm,  . N  ! ( ()L M M M 66GN9- -0CI0N N N  " " $ $ $(,GN9 %$&G ! ! !)''" O Nrcxt|ddx}r|j|jk}t|jdz |jdkrkt jtdtd|j j | |j |j|t|jd<dSdSdS)z?Upload model checkpoints to Ultralytics HUB with rate limiting.rNrzUploading checkpoint z/models/)r best_fitnessfitnessrrr-rinforrrid upload_modelr&last)rris_bests r on_model_saver91s'=$777w,&'/9 66GN6* *W-@-H H H K6`` ``gmN^`` a a a   g F F F%)VVGN6 " " " ,, I HrcFt|ddx}rtjtd||j|j|jdddd|_ tjtd td |j d dSdS) zIUpload final model and metrics to Ultralytics HUB at the end of training.rNzSyncing final model...zmetrics/mAP50-95(B)rT)mapfinalFu Done ✅ zView model at u 🚀) rrr4rr6r&bestrgetalive model_urlrs r on_train_endrA<s'=$777w Y v555666 M L##$91==      vWWWWw?PWWWXXXXX Y Yrc:t|j|jdS)zRun events on train start.Nr rdevicers ron_train_startrEKs 7<(((((rcL|jst|j|jdSdS)zRun events on validation start.N)trainingr rrD) validators r on_val_startrIPs2  1y~y/0000011rc~tt|dddd}t|j|j|dS)zRun events on predict start.rNbackend)rK)rr rrD) predictorrKs ron_predict_startrMVs?gi$77DIIG 9>9+W======rc:t|j|jdS)zRun events on export start.NrC)exporters ron_export_startrP\s 8=(/*****rr T) rrr0r9rArErIrMrP)r(rultralytics.hubrrrultralytics.utilsrrr ultralytics.utils.eventsr rrr0r9rArErIrMrP callbacksrrrVsH DDDDDDDDDD4444444444++++++bbb ==='''2,,, Y Y Y))) 111 >>> +++"$&?#:,&$($,*     r