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

ultralytics 8.3.20 W&B plots=False logging fix (#17088)

Browse source 
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Anzhc 2024-10-22 19:30:20 +03:00 • committed by GitHub
parent 901b68aa5c
commit f4d8f7765a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 14 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/__init__.py
View file

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, AGPL-3.0 license # Ultralytics YOLO 🚀, AGPL-3.0 license
__version__ = "8.3.19" __version__ = "8.3.20"
import os import os

24
ultralytics/utils/callbacks/wb.py
View file

@ -137,17 +137,19 @@ def on_train_end(trainer):
if trainer.best.exists(): if trainer.best.exists():
art.add_file(trainer.best) art.add_file(trainer.best)
wb.run.log_artifact(art, aliases=["best"]) wb.run.log_artifact(art, aliases=["best"])
for curve_name, curve_values in zip(trainer.validator.metrics.curves, trainer.validator.metrics.curves_results): # Check if we actually have plots to save
x, y, x_title, y_title = curve_values if trainer.args.plots:
_plot_curve( for curve_name, curve_values in zip(trainer.validator.metrics.curves, trainer.validator.metrics.curves_results):
x, x, y, x_title, y_title = curve_values
y, _plot_curve(
names=list(trainer.validator.metrics.names.values()), x,
id=f"curves/{curve_name}", y,
title=curve_name, names=list(trainer.validator.metrics.names.values()),
x_title=x_title, id=f"curves/{curve_name}",
y_title=y_title, title=curve_name,
) x_title=x_title,
y_title=y_title,
)
wb.run.finish() # required or run continues on dashboard wb.run.finish() # required or run continues on dashboard