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