From 9c6989a1b0d0b94c51eacf7cb123c028145b224e Mon Sep 17 00:00:00 2001 From: Maksym Shamrai Date: Wed, 29 Nov 2023 03:29:15 +0200 Subject: [PATCH] `ultralytics 8.0.220` fix UnboundLocalError in tuner.py (#6614) Co-authored-by: Glenn Jocher --- ultralytics/__init__.py | 2 +- ultralytics/engine/tuner.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index 710a7723..bb102a66 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = '8.0.219' +__version__ = '8.0.220' from ultralytics.models import RTDETR, SAM, YOLO from ultralytics.models.fastsam import FastSAM diff --git a/ultralytics/engine/tuner.py b/ultralytics/engine/tuner.py index 6d6e0a86..1e5090ab 100644 --- a/ultralytics/engine/tuner.py +++ b/ultralytics/engine/tuner.py @@ -175,12 +175,12 @@ class Tuner: metrics = {} train_args = {**vars(self.args), **mutated_hyp} save_dir = get_save_dir(get_cfg(train_args)) + weights_dir = save_dir / 'weights' + ckpt_file = weights_dir / ('best.pt' if (weights_dir / 'best.pt').exists() else 'last.pt') try: # Train YOLO model with mutated hyperparameters (run in subprocess to avoid dataloader hang) - weights_dir = save_dir / 'weights' cmd = ['yolo', 'train', *(f'{k}={v}' for k, v in train_args.items())] assert subprocess.run(cmd, check=True).returncode == 0, 'training failed' - ckpt_file = weights_dir / ('best.pt' if (weights_dir / 'best.pt').exists() else 'last.pt') metrics = torch.load(ckpt_file)['train_metrics'] except Exception as e: