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
4
.github/workflows/ci.yaml
vendored
4
.github/workflows/ci.yaml
vendored
|
|
@ -184,7 +184,7 @@ jobs:
|
||||||
torch="torch==1.8.0 torchvision==0.9.0"
|
torch="torch==1.8.0 torchvision==0.9.0"
|
||||||
fi
|
fi
|
||||||
if [[ "${{ github.event_name }}" =~ ^(schedule|workflow_dispatch)$ ]]; then
|
if [[ "${{ github.event_name }}" =~ ^(schedule|workflow_dispatch)$ ]]; then
|
||||||
slow="pycocotools mlflow ray[tune]"
|
slow="pycocotools mlflow"
|
||||||
fi
|
fi
|
||||||
pip install -e ".[export]" $torch $slow pytest-cov --extra-index-url https://download.pytorch.org/whl/cpu
|
pip install -e ".[export]" $torch $slow pytest-cov --extra-index-url https://download.pytorch.org/whl/cpu
|
||||||
- name: Check environment
|
- name: Check environment
|
||||||
|
|
@ -247,7 +247,7 @@ jobs:
|
||||||
- name: Install requirements
|
- name: Install requirements
|
||||||
run: |
|
run: |
|
||||||
python -m pip install --upgrade pip wheel
|
python -m pip install --upgrade pip wheel
|
||||||
pip install -e ".[export]" pytest mlflow pycocotools "ray[tune]"
|
pip install -e ".[export]" pytest mlflow pycocotools
|
||||||
- name: Check environment
|
- name: Check environment
|
||||||
run: |
|
run: |
|
||||||
yolo checks
|
yolo checks
|
||||||
|
|
|
||||||
|
|
@ -1,12 +1,16 @@
|
||||||
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
||||||
|
|
||||||
|
|
||||||
from ultralytics.cfg import TASK2DATA, TASK2METRIC, get_save_dir
|
from ultralytics.cfg import TASK2DATA, TASK2METRIC, 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
|
||||||
|
|
||||||
|
|
||||||
def run_ray_tune(
|
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.
|
Runs hyperparameter tuning using Ray Tune.
|
||||||
|
|
@ -38,7 +42,7 @@ def run_ray_tune(
|
||||||
train_args = {}
|
train_args = {}
|
||||||
|
|
||||||
try:
|
try:
|
||||||
checks.check_requirements(("ray[tune]", "numpy<2.0.0"))
|
checks.check_requirements("ray[tune]")
|
||||||
|
|
||||||
import ray
|
import ray
|
||||||
from ray import tune
|
from ray import tune
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue