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

logger updates (#97)

Browse Source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Ayush Chaurasia
2022-12-26 22:42:17 +05:30
committed by GitHub
parent 48cffa176e
commit a1808eeda4
5 changed files with 58 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ultralytics/yolo/utils/callbacks/clearml.py
View File

@ -16,7 +16,7 @@ def _log_images(imgs_dict, group="", step=0):
task.get_logger().report_image(group, k, step, v)
def on_train_start(trainer):
def on_pretrain_routine_start(trainer):
# TODO: reuse existing task
task = Task.init(project_name=trainer.args.project if trainer.args.project != 'runs/train' else 'YOLOv8',
task_name=trainer.args.name,
@ -48,7 +48,7 @@ def on_train_end(trainer):
callbacks = {
"on_train_start": on_train_start,
"on_pretrain_routine_start": on_pretrain_routine_start,
"on_train_epoch_end": on_train_epoch_end,
"on_val_end": on_val_end,
"on_train_end": on_train_end} if clearml else {}