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

7
ultralytics/yolo/utils/callbacks/tb.py
View file

@ -8,7 +8,7 @@ def _log_scalars(scalars, step=0):
writer.add_scalar(k, v, step)
def on_train_start(trainer):
def on_pretrain_routine_start(trainer):
global writer
writer = SummaryWriter(str(trainer.save_dir))
@ -21,4 +21,7 @@ def on_val_end(trainer):
_log_scalars(trainer.metrics, trainer.epoch + 1)
callbacks = {"on_train_start": on_train_start, "on_val_end": on_val_end, "on_batch_end": on_batch_end}
callbacks = {
"on_pretrain_routine_start": on_pretrain_routine_start,
"on_val_end": on_val_end,
"on_batch_end": on_batch_end}