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

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

@ -137,6 +137,8 @@ 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"])
# Check if we actually have plots to save
if trainer.args.plots:
for curve_name, curve_values in zip(trainer.validator.metrics.curves, trainer.validator.metrics.curves_results): for curve_name, curve_values in zip(trainer.validator.metrics.curves, trainer.validator.metrics.curves_results):
x, y, x_title, y_title = curve_values x, y, x_title, y_title = curve_values
_plot_curve( _plot_curve(