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

Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
pull/17097/head v8.3.20
Anzhc 1 month ago committed by GitHub
parent 901b68aa5c
commit f4d8f7765a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 2
      ultralytics/__init__.py
  2. 24
      ultralytics/utils/callbacks/wb.py

@ -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

@ -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

Loading…
Cancel
Save