Disable Ray tests (#17266)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
b8c90baffe
commit
11b4194344
2 changed files with 9 additions and 5 deletions
|
|
@ -1,12 +1,16 @@
|
|||
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
||||
|
||||
|
||||
from ultralytics.cfg import TASK2DATA, TASK2METRIC, get_save_dir
|
||||
from ultralytics.utils import DEFAULT_CFG, DEFAULT_CFG_DICT, LOGGER, NUM_THREADS, checks
|
||||
|
||||
|
||||
def run_ray_tune(
|
||||
model, space: dict = None, grace_period: int = 10, gpu_per_trial: int = None, max_samples: int = 10, **train_args
|
||||
model,
|
||||
space: dict = None,
|
||||
grace_period: int = 10,
|
||||
gpu_per_trial: int = None,
|
||||
max_samples: int = 10,
|
||||
**train_args,
|
||||
):
|
||||
"""
|
||||
Runs hyperparameter tuning using Ray Tune.
|
||||
|
|
@ -38,7 +42,7 @@ def run_ray_tune(
|
|||
train_args = {}
|
||||
|
||||
try:
|
||||
checks.check_requirements(("ray[tune]", "numpy<2.0.0"))
|
||||
checks.check_requirements("ray[tune]")
|
||||
|
||||
import ray
|
||||
from ray import tune
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue