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

ultralytics 8.0.177 add https://youtube.com/ultralytics videos to Docs (#4875)

Browse source 
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: Muhammad Rizwan Munawar <62513924+RizwanMunawar@users.noreply.github.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2023-09-13 20:42:50 +02:00 • committed by GitHub
parent e73447effb
commit dd2262e89a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
30 changed files with 453 additions and 161 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/conftest.py
View file

@ -29,18 +29,14 @@ def pytest_configure(config):
config.addinivalue_line('markers', 'slow: mark test as slow to run')
def pytest_collection_modifyitems(config, items):
"""Modify collected test items based on custom command-line options.
def pytest_runtest_setup(item):
"""Setup hook to skip tests marked as slow if the --slow option is not provided.
Args:
config (pytest.config.Config): The pytest config object.
items (list): List of collected test items.
item (pytest.Item): The test item object.
"""
if not config.getoption('--slow'):
skip_slow = pytest.mark.skip(reason="remove this test because it's slow")
for item in items:
if 'slow' in item.keywords:
item.add_marker(skip_slow)
if 'slow' in item.keywords and not item.config.getoption('--slow'):
pytest.skip('skip slow tests unless --slow is set')
def pytest_sessionstart(session):

6
tests/test_cli.py
View file

@ -114,9 +114,9 @@ def test_mobilesam():
# model(source)
# Slow Tests
# Slow Tests -----------------------------------------------------------------------------------------------------------
@pytest.mark.slow
@pytest.mark.parametrize('task,model,data', TASK_ARGS)
def test_train_gpu(task, model, data):
run(f'yolo train {task} model={model}.yaml data={data} imgsz=32 epochs=1 device="0"') # single GPU
run(f'yolo train {task} model={model}.pt data={data} imgsz=32 epochs=1 device="0,1"') # multi GPU
run(f'yolo train {task} model={model}.yaml data={data} imgsz=32 epochs=1 device=0') # single GPU
run(f'yolo train {task} model={model}.pt data={data} imgsz=32 epochs=1 device=0,1') # multi GPU