ultralytics 8.3.20 W&B plots=False logging fix (#17088)
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
parent
901b68aa5c
commit
f4d8f7765a
2 changed files with 14 additions and 12 deletions
|
|
@ -1,6 +1,6 @@
|
|||
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
||||
|
||||
__version__ = "8.3.19"
|
||||
__version__ = "8.3.20"
|
||||
|
||||
import os
|
||||
|
||||
|
|
|
|||
|
|
@ -137,6 +137,8 @@ def on_train_end(trainer):
|
|||
if trainer.best.exists():
|
||||
art.add_file(trainer.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):
|
||||
x, y, x_title, y_title = curve_values
|
||||
_plot_curve(
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue