HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

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
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
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 {}