HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Allow tuning directory to be set using name (#18760)

Browse source 
Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Mohammed Yasin 2025-01-20 00:33:02 +08:00 • committed by GitHub
parent 42a220c71d
commit a8e2464a9c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
ultralytics/engine/tuner.py
View file

@ -101,7 +101,8 @@ class Tuner:
"copy_paste": (0.0, 1.0), # segment copy-paste (probability) "copy_paste": (0.0, 1.0), # segment copy-paste (probability)
} }
self.args = get_cfg(overrides=args) self.args = get_cfg(overrides=args)
self.tune_dir = get_save_dir(self.args, name="tune") self.tune_dir = get_save_dir(self.args, name=self.args.name or "tune")
self.args.name = None # reset to not affect training directory
self.tune_csv = self.tune_dir / "tune_results.csv" self.tune_csv = self.tune_dir / "tune_results.csv"
self.callbacks = _callbacks or callbacks.get_default_callbacks() self.callbacks = _callbacks or callbacks.get_default_callbacks()
self.prefix = colorstr("Tuner: ") self.prefix = colorstr("Tuner: ")

6
ultralytics/utils/tuner.py
View file

@ -1,6 +1,6 @@
# Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license
from ultralytics.cfg import TASK2DATA, TASK2METRIC, get_save_dir from ultralytics.cfg import TASK2DATA, TASK2METRIC, get_cfg, get_save_dir
from ultralytics.utils import DEFAULT_CFG, DEFAULT_CFG_DICT, LOGGER, NUM_THREADS, checks from ultralytics.utils import DEFAULT_CFG, DEFAULT_CFG_DICT, LOGGER, NUM_THREADS, checks
@ -134,7 +134,9 @@ def run_ray_tune(
tuner_callbacks = [WandbLoggerCallback(project="YOLOv8-tune")] if wandb else [] tuner_callbacks = [WandbLoggerCallback(project="YOLOv8-tune")] if wandb else []
# Create the Ray Tune hyperparameter search tuner # Create the Ray Tune hyperparameter search tuner
tune_dir = get_save_dir(DEFAULT_CFG, name="tune").resolve() # must be absolute dir tune_dir = get_save_dir(
get_cfg(DEFAULT_CFG, train_args), name=train_args.pop("name", "tune")
).resolve() # must be absolute dir
tune_dir.mkdir(parents=True, exist_ok=True) tune_dir.mkdir(parents=True, exist_ok=True)
tuner = tune.Tuner( tuner = tune.Tuner(
trainable_with_resources, trainable_with_resources,