xdobro23/YOLOv8-16bit
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Threadpool fixes and CLI improvements (#550)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Ayush Chaurasia <ayush.chaurarsia@gmail.com>
This commit is contained in:
Glenn Jocher
2023-01-22 17:08:08 +01:00
committed by GitHub
parent d9a0fba251
commit 21b701c4ea
22 changed files with 338 additions and 251 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ultralytics/yolo/utils/callbacks/hub.py
View File

@ -3,7 +3,7 @@
import json
from time import time
from ultralytics.hub.utils import PREFIX, sync_analytics
from ultralytics.hub.utils import PREFIX, traces
from ultralytics.yolo.utils import LOGGER
@ -43,24 +43,24 @@ def on_train_end(trainer):
LOGGER.info(f"{PREFIX}Training completed successfully ✅\n"
f"{PREFIX}Uploading final {session.model_id}")
session.upload_model(trainer.epoch, trainer.best, map=trainer.metrics['metrics/mAP50-95(B)'], final=True)
session.alive = False # stop heartbeats
session.shutdown() # stop heartbeats
LOGGER.info(f"{PREFIX}View model at https://hub.ultralytics.com/models/{session.model_id} 🚀")
def on_train_start(trainer):
sync_analytics(trainer.args)
traces(trainer.args, traces_sample_rate=0.0)
def on_val_start(validator):
sync_analytics(validator.args)
traces(validator.args, traces_sample_rate=0.0)
def on_predict_start(predictor):
sync_analytics(predictor.args)
traces(predictor.args, traces_sample_rate=0.0)
def on_export_start(exporter):
sync_analytics(exporter.args)
traces(exporter.args, traces_sample_rate=0.0)
callbacks = {